| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 47 std::vector<int64> broadcast_shape; in BroadcastTo() local
|
| /external/tensorflow/tensorflow/contrib/coder/kernels/ |
| D | range_coder_ops_util.cc | 28 Status MergeAxes(const TensorShape& broadcast_shape, in MergeAxes()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_broadcast_helper_op.cc | 81 std::vector<int64> broadcast_shape(max_rank_shape->dims(), 1LL); in Compile() local
|
| D | diag_op.cc | 71 auto broadcast_shape = in CreateDiagonal() local
|
| D | image_resize_ops.cc | 215 xla::Shape broadcast_shape = broadcast_shape_or_status.ValueOrDie(); in BroadcastSpatialDimensions() local
|
| /external/tensorflow/tensorflow/contrib/gan/python/features/python/ |
| D | virtual_batchnorm_impl.py | 256 def _broadcast(self, v, broadcast_shape=None): argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | common_shapes.py | 578 def broadcast_shape(shape_x, shape_y): function
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_creation_utils.cc | 180 Shape broadcast_shape = ShapeUtil::MakeShape(operand->shape().element_type(), in MakeBroadcastHlo() local
|
| D | tuple_points_to_analysis_test.cc | 1261 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {16}); in TEST_F() local
|
| D | hlo_instruction.cc | 1122 Shape broadcast_shape = ShapeUtil::ChangeElementType( in CreateBroadcastSequence() local
|
| D | algebraic_simplifier_test.cc | 3511 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {4, 5, 6, 7}); in TEST_F() local
3544 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {4, 5, 6}); in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 336 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {1, 8, 8}); in TEST_F() local
|
| /external/tensorflow/tensorflow/contrib/rnn/kernels/ |
| D | lstm_ops.cc | 67 Eigen::array<Eigen::DenseIndex, 2> broadcast_shape({cell.batch_size(), 1}); in LSTMBlockCellFpropWithEigen() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 2637 TensorShapeProto broadcast_shape; in TrySimplifyInternal() local
2723 TensorShapeProto broadcast_shape; in TrySimplify() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 728 int broadcast_shape[5]; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 752 int broadcast_shape[5]; member
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 439 Shape broadcast_shape = in AddBroadcastSequence() local
|