Home
last modified time | relevance | path

Searched defs:broadcast_shape (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/python/framework/
Dcommon_shapes.py89 def broadcast_shape(shape_x, shape_y): function
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dbroadcast.cc50 std::vector<int64> broadcast_shape; in BroadcastTo() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_broadcast_helper_op.cc81 std::vector<int64> broadcast_shape(max_rank_shape->dims(), 1LL); in Compile() local
Dimage_resize_ops.cc220 xla::Shape broadcast_shape = broadcast_shape_or_status.ValueOrDie(); in BroadcastSpatialDimensions() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc196 Shape broadcast_shape = ShapeUtil::MakeShape(operand->shape().element_type(), in MakeBroadcastHlo() local
Dhlo_instruction.cc1225 Shape broadcast_shape = ShapeUtil::ChangeElementType( in CreateBroadcastSequence() local
Dalgebraic_simplifier_test.cc3900 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {4, 5, 6, 7}); in TEST_F() local
3933 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {4, 5, 6}); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc338 Shape broadcast_shape = ShapeUtil::MakeShape(F32, {1, 8, 8}); in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/rnn/
Dlstm_ops.cc67 Eigen::array<Eigen::DenseIndex, 2> broadcast_shape({cell.batch_size(), 1}); in LSTMBlockCellFpropWithEigen() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc2707 TensorShapeProto broadcast_shape; in TrySimplifyInternal() local
2776 TensorShapeProto broadcast_shape; in TrySimplify() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h784 int broadcast_shape[5]; member
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc1313 llvm::SmallVector<int64_t, 4> broadcast_shape; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc494 Shape broadcast_shape = in AddBroadcastSequence() local