Home
last modified time | relevance | path

Searched defs:broadcast_dimensions (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc615 absl::Span<const int64_t> broadcast_dimensions) { in InDimBroadcast()
644 std::vector<int64_t> broadcast_dimensions; in AddBroadcastSequence() local
689 absl::Span<const int64_t> broadcast_dimensions, in BinaryOp()
958 const absl::Span<const int64_t> broadcast_dimensions) { in BroadcastInDim()
4060 const absl::Span<const int64_t> broadcast_dimensions) { in BroadcastInDim()
4151 absl::Span<const int64_t> broadcast_dimensions) { in Eq()
4156 absl::Span<const int64_t> broadcast_dimensions, in CompareTotalOrder()
4172 absl::Span<const int64_t> broadcast_dimensions) { in EqTotalOrder()
4178 absl::Span<const int64_t> broadcast_dimensions) { in Ne()
4183 absl::Span<const int64_t> broadcast_dimensions) { in NeTotalOrder()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddiag_op.cc68 std::vector<int64_t> broadcast_dimensions(other_dims.size() + 1); in CreateDiagonal() local
Dselect_op.cc69 std::vector<int64_t> broadcast_dimensions(cond_shape.dims()); in Compile() local
Dmatrix_diag_ops.cc161 std::vector<int64_t> broadcast_dimensions(input_rank - 1); in SetMatrixDiag() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc238 absl::Span<const int64_t> broadcast_dimensions, in MakeBroadcastHlo()
252 absl::Span<const int64_t> broadcast_dimensions, in MakeBroadcastHlo()
670 absl::Span<const int64_t> broadcast_dimensions) { in BroadcastZeros()
679 absl::Span<const int64_t> broadcast_dimensions) { in BroadcastOnes()
Dshape_inference.cc784 absl::Span<const int64_t> broadcast_dimensions) { in InferInDimBroadcastShape()
901 absl::Span<const int64_t> broadcast_dimensions) { in InferElementwiseBinaryOpShape()
963 absl::Span<const int64_t> broadcast_dimensions) { in InferBinaryOpShape()
3001 absl::Span<const int64_t> broadcast_dimensions) { in InferBroadcastShape()
Dhlo_instruction.cc1662 absl::Span<const int64_t> broadcast_dimensions) { in CreateBroadcast()
1704 std::vector<int64_t> broadcast_dimensions; in CreateBroadcastSequence() local
Dhlo_parser.cc2114 optional<std::vector<int64_t>> broadcast_dimensions; in CreateInstruction() local
Dalgebraic_simplifier.cc3781 std::vector<int64_t> broadcast_dimensions; in HandlePad() local
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc513 absl::Span<const int64_t> broadcast_dimensions) { in InDimBroadcast()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo.cc3269 bool IsTFStyleBroadcast(DenseIntElementsAttr broadcast_dimensions, in IsTFStyleBroadcast()
3283 DenseIntElementsAttr broadcast_dimensions, in ExpandedShape()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc327 llvm::SmallVector<int64_t, 4> broadcast_dimensions; in GetStaticBroadcastType() local
1624 llvm::SmallVector<int64_t, 4> broadcast_dimensions; in matchAndRewrite() local
4653 SmallVector<int64_t, 4> broadcast_dimensions; in matchAndRewrite() local
4763 SmallVector<int64_t, 4> broadcast_dimensions; in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc176 llvm::Optional<mlir::DenseIntElementsAttr> broadcast_dimensions) { in Convert_broadcast_dimensions()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1202 llvm::SmallVector<int64_t> broadcast_dimensions; in RowVectorizationEnabled() local