Searched defs:broadcast_shape (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/contrib/coder/kernels/ |
D | range_coder_ops_util.cc | 28 Status MergeAxes(const TensorShape& broadcast_shape, in MergeAxes()
|
/external/tensorflow/tensorflow/contrib/gan/python/features/python/ |
D | virtual_batchnorm_impl.py | 255 def _broadcast(self, v, broadcast_shape=None): argument
|
/external/tensorflow/tensorflow/python/framework/ |
D | common_shapes.py | 566 def broadcast_shape(shape_x, shape_y): function
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | diag_op.cc | 75 auto broadcast_shape = in CreateDiagonal() local
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_instruction_fusion_test.cc | 274 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {1, 8, 8}); in TEST_F() local
|
/external/tensorflow/tensorflow/contrib/rnn/kernels/ |
D | lstm_ops.cc | 66 Eigen::array<Eigen::DenseIndex, 2> broadcast_shape({cell.batch_size(), 1}); in LSTMBlockCellFpropWithEigen() local
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | algebraic_simplifier_test.cc | 2161 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {4, 5, 6, 7}); in TEST_F() local 2198 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {4, 5, 6}); in TEST_F() local
|
D | user_computation.cc | 3442 Shape broadcast_shape = ShapeUtil::MakeShape( in Visit() local
|
D | hlo_instruction.cc | 682 Shape broadcast_shape = ShapeUtil::ChangeElementType( in CreateBroadcastSequence() local
|