Searched defs:broadcast_dims (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dcwise_op_gpu_select.cu.cc49 Eigen::array<int, 1> broadcast_dims{size};
52 .broadcast(broadcast_dims)
68 Eigen::array<int, 2> broadcast_dims{{ 1, all_but_batch }};
71 Eigen::IndexList<Eigen::type2index<1>, int> broadcast_dims; local
72 broadcast_dims.set(1, all_but_batch);
82 // .broadcast(broadcast_dims)
87 .broadcast(broadcast_dims)
H A Dadjust_contrast_op.h47 Eigen::array<int, 4> broadcast_dims; local
48 broadcast_dims[0] = 1;
49 broadcast_dims[1] = height;
50 broadcast_dims[2] = width;
51 broadcast_dims[3] = 1;
61 broadcast_dims; local
62 broadcast_dims.set(1, height);
63 broadcast_dims.set(2, width);
75 .broadcast(broadcast_dims);
109 Eigen::array<int, 4> broadcast_dims; local
128 broadcast_dims; local
[all...]
H A Dgather_nd_op_cpu_impl.h108 Eigen::array<Eigen::DenseIndex, 1> broadcast_dims{{ batch_size }};
111 Eigen::IndexList<Eigen::DenseIndex> broadcast_dims; local
112 broadcast_dims.set(0, batch_size);
117 .broadcast(broadcast_dims)
H A Dcwise_op_select.cc241 Eigen::array<Eigen::DenseIndex, 2> broadcast_dims{{1, all_but_batch}};
244 Eigen::IndexList<Eigen::type2index<1>, Eigen::DenseIndex> broadcast_dims; local
245 broadcast_dims.set(1, all_but_batch);
252 .broadcast(broadcast_dims)
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier_test.cc1624 const std::vector<int64> broadcast_dims = local
1626 EXPECT_EQ(1, broadcast_dims.size());
1627 EXPECT_THAT(broadcast_dims[0], ::testing::AnyOf(1, 2, 3));

Completed in 156 milliseconds