| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_op_gpu_select.cu.cc | 66 Eigen::array<int, 1> broadcast_dims{size}; in operator ()() local
88 Eigen::IndexList<Eigen::type2index<1>, int> broadcast_dims; in operator ()() local
|
| D | cwise_op_select.cc | 361 Eigen::IndexList<Eigen::type2index<1>, Eigen::DenseIndex> broadcast_dims; in operator ()() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | adjust_contrast_op.h | 47 broadcast_dims; in operator() local
95 broadcast_dims; in operator() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | broadcast.cc | 45 std::vector<int64_t> broadcast_dims; in BroadcastTo() local
|
| D | svd.cc | 138 std::vector<int64_t> broadcast_dims(num_dims - 1); in HouseRow() local
203 std::vector<int64_t> broadcast_dims(num_dims - 1); in HouseCol() local
480 std::vector<int64_t> broadcast_dims(batch_dims.size()); in OneSidedJacobiUpdate() local
742 std::vector<int64_t> broadcast_dims(num_dims - 1); in SortBySingularValuesAndPostProcessing() local
|
| D | svd_test.cc | 92 std::vector<int64_t> broadcast_dims(num_dims - 1); in ComputeMatmulUDVT() local
|
| D | matrix.cc | 183 std::vector<int64_t> broadcast_dims(n_dims - 1); in SetMatrixDiagonal() local
274 std::vector<int64_t> broadcast_dims; in EinsumDiagonalLabels() local
|
| D | self_adjoint_eig_test.cc | 122 std::vector<int64_t> broadcast_dims(shape.rank() - 1); in ComputeMatmulVWVt() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_broadcast_helper_op.cc | 45 std::vector<int64_t> broadcast_dims; in Compile() local
|
| D | resampler_ops.cc | 62 std::vector<int64_t> broadcast_dims(warp_dims.begin(), warp_dims.end() - 1); in BilinearWeights() local
|
| D | image_ops.cc | 214 std::vector<int64_t> broadcast_dims(input_shape.dims() - 2); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | eigh_expander.cc | 150 std::vector<int64_t> broadcast_dims(shape.dimensions().size() - 1); in ApplyJacobiRotationOverRows() local
170 std::vector<int64_t> broadcast_dims(shape.dimensions().size() - 1); in ApplyJacobiRotationOverCols() local
361 std::vector<int64_t> broadcast_dims(num_dims - 1); in SortByEigenvalues() local
|
| D | convolution_group_converter.cc | 468 std::vector<int64_t> broadcast_dims; in HandleConvolution() local
|
| D | space_to_batch_converter.cc | 1444 std::vector<int64_t> broadcast_dims; in PropagateOnBroadcast() local
1519 auto broadcast_dims = broadcast->dimensions(); in IsBroadcastPropagatable() local
2351 std::vector<int64_t> broadcast_dims(new_space_dims.begin(), in SelectValidPortion() local
|
| D | while_loop_concat_code_motion.cc | 872 std::vector<int64_t> broadcast_dims; in RewriteLoopWithConcatGroups() local
|
| D | indexed_array_analysis.cc | 880 absl::Span<const int64_t> broadcast_dims = broadcast_instr->dimensions(); in ComputeArrayForElementwiseBinaryOp() local
|
| D | dynamic_padder.cc | 1481 std::vector<int64_t> broadcast_dims; in RewriteDynamicBinaryOp() local
|
| D | algebraic_simplifier.cc | 1525 DimensionVector broadcast_dims; in HandleConcatenate() local
5707 DimensionVector broadcast_dims(nested_root->shape().dimensions_size()); in HandleReduceWindow() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_rewriter.cc | 327 absl::Span<const int64_t> broadcast_dims = broadcast_bias->dimensions(); in FuseVectorBiasAdd() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 96 std::vector<int64_t> broadcast_dims(indices_shape.dims()); in OneHot() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_collective.cc | 167 auto broadcast_dims = GetI64ElementsAttr({}, &builder); in ConvertAllReduce() local
|
| D | legalize_tf.cc | 354 DenseIntElementsAttr broadcast_dims, in StaticBinaryBroadcast()
410 auto broadcast_dims = GetI64ElementsAttr({feature_dim}, &builder); in Broadcast1DToFeatureDim() local
435 auto broadcast_dims = GetI64ElementsAttrForSeq(0, rank, &builder); in BroadcastToShapeOf() local
4182 auto broadcast_dims = GetI64ElementsAttr({}, &rewriter); in matchAndRewrite() local
5311 llvm::SmallVector<int64_t, 4> broadcast_dims(indices_shape.size()); in matchAndRewrite() local
6144 auto broadcast_dims = GetI64ElementsAttr({}, &rewriter); in matchAndRewrite() local
|
| /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/xla/service/spmd/ |
| D | spmd_partitioner.cc | 261 absl::flat_hash_set<int64_t> broadcast_dims; in AddInstruction() local
1184 std::vector<int64_t> broadcast_dims; in ReplicatePartial() local
1954 if (auto broadcast_dims = state_.b->BroadcastDimsForCreatedHlo(hlo())) { in ReshardWithCollectivePermute() local
|