Searched defs:broadcast_dims (Results 1 - 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | cwise_op_gpu_select.cu.cc | 49 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 D | adjust_contrast_op.h | 47 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 D | gather_nd_op_cpu_impl.h | 108 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 D | cwise_op_select.cc | 241 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 D | algebraic_simplifier_test.cc | 1624 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