| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | adjust_contrast_op.h | 47 Eigen::array<int, 4> broadcast_dims; in operator() local
61 broadcast_dims; in operator() local
109 Eigen::array<int, 4> broadcast_dims; in operator() local
128 broadcast_dims; in operator() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_gpu_select.cu.cc | 65 Eigen::array<int, 1> broadcast_dims{size}; in operator ()() local
84 Eigen::array<int, 2> broadcast_dims{{ 1, all_but_batch }}; in operator ()() local
87 Eigen::IndexList<Eigen::type2index<1>, int> broadcast_dims; in operator ()() local
|
| D | cwise_op_select.cc | 341 Eigen::array<Eigen::DenseIndex, 2> broadcast_dims{{1, all_but_batch}}; in operator ()() local
344 Eigen::IndexList<Eigen::type2index<1>, Eigen::DenseIndex> broadcast_dims; in operator ()() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 49 std::vector<int64> broadcast_dims; in BroadcastTo() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_broadcast_helper_op.cc | 45 std::vector<int64> broadcast_dims; in Compile() local
|
| D | resampler_ops.cc | 60 std::vector<int64> broadcast_dims(warp_dims.begin(), warp_dims.end() - 1); in BilinearWeights() local
|
| D | image_ops.cc | 207 std::vector<int64> broadcast_dims(input_shape.dims() - 2); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
| D | broadcast_utils.cc | 32 DenseIntElementsAttr broadcast_dims) { in IsLegalNumpyRankedBroadcast()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd.cc | 137 std::vector<int64> broadcast_dims(num_dims - 1); in HouseRow() local
202 std::vector<int64> broadcast_dims(num_dims - 1); in HouseCol() local
479 std::vector<int64> broadcast_dims(batch_dims.size()); in OneSidedJacobiUpdate() local
727 std::vector<int64> broadcast_dims(num_dims - 1); in SortBySingularValuesAndPostProcessing() local
|
| D | self_adjoint_eig.cc | 165 std::vector<int64> broadcast_dims(batch_dims.size()); in Update() local
346 std::vector<int64> broadcast_dims(num_dims - 1); in SortByEigenvalues() local
|
| D | matrix.cc | 182 std::vector<int64> broadcast_dims(n_dims - 1); in SetMatrixDiagonal() local
243 std::vector<int64> broadcast_dims; in EinsumDiagonalLabels() local
|
| D | svd_test.cc | 92 std::vector<int64> broadcast_dims(num_dims - 1); in ComputeMatmulUDVT() local
|
| D | self_adjoint_eig_test.cc | 106 std::vector<int64> broadcast_dims(shape.rank() - 1); in ComputeMatmulVWVt() local
|
| D | arithmetic.cc | 194 std::vector<int64> broadcast_dims(input_shape.rank() - 1); in ArgMinMaxTwoPass() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 88 std::vector<int64> broadcast_dims(indices_shape.dims()); in OneHot() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | convolution_group_converter.cc | 462 std::vector<int64> broadcast_dims; in HandleConvolution() local
|
| D | space_to_batch_converter.cc | 1034 std::vector<int64> broadcast_dims; in PropagateOnBroadcast() local
1107 auto broadcast_dims = broadcast->dimensions(); in IsBroadcastPropagatable() local
|
| D | indexed_array_analysis.cc | 876 absl::Span<const int64> broadcast_dims = broadcast_instr->dimensions(); in ComputeArrayForElementwiseBinaryOp() local
|
| D | algebraic_simplifier.cc | 1355 absl::InlinedVector<int64, 8> broadcast_dims; in HandleConcatenate() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | legalize_to_linalg.cc | 547 auto broadcast_dims = std::get<1>(operand_and_dims); in matchAndRewrite() local
606 SmallVector<int64_t, 2> new_shape, new_strides, broadcast_dims; in InsertReshapeIfNecessary() local
657 ArrayRef<int64_t> broadcast_dims, in getIndexingMaps()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 122 def broadcasting_binary_op_wrapper(x, y, broadcast_dims=None, name=None): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 425 DenseIntElementsAttr broadcast_dims, in StaticBinaryBroadcast()
508 auto broadcast_dims = GetI64ElementsAttr({feature_dim}, &builder); in Broadcast1DToFeatureDim() local
533 auto broadcast_dims = GetI64ElementsAttrForSeq(0, rank, &builder); in BroadcastToShapeOf() local
3685 auto broadcast_dims = GetI64ElementsAttr({}, &rewriter); in matchAndRewrite() local
4506 llvm::SmallVector<int64_t, 4> broadcast_dims(indices_shape.size()); in matchAndRewrite() local
5303 auto broadcast_dims = GetI64ElementsAttr({}, &rewriter); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 231 absl::flat_hash_set<int64> broadcast_dims; in AddInstruction() local
1317 if (auto broadcast_dims = state_.b->BroadcastDimsForCreatedHlo(hlo())) { in ReshardWithCollectivePermute() local
|