| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | slice_spmd_expander.cc | 387 const int output_rank, in ApplyNewAndShrinkMasksToLayout()
616 const int output_rank = ValueRank(strided_slice_op.output()); in ExpandOp() local
656 const int output_rank = ValueRank(strided_slice_op.output()); in ComputeLayoutForward() local
682 const int output_rank = ValueRank(strided_slice_op.output()); in ComputeLayoutBackward() local
873 const int output_rank = global_output_shape.size(); in ExpandOp() local
940 const int output_rank = global_output_shape.size(); in ComputeLayoutForward() local
967 const int output_rank = global_output_shape.size(); in ComputeLayoutBackward() local
|
| D | segmentation_spmd_expander.cc | 38 const int output_rank = ValueRank(unsorted_segmented_sum.output()); in ComputeLayoutForward() local
|
| D | elementwise_spmd_expander.cc | 129 const int output_rank = ValueRank(op->getOpResult(0)); in ComputeLayoutForward() local
|
| D | meta_spmd_expander.cc | 136 const int output_rank = ValueRank(pack.output()); in ExpandOp() local
1235 const int output_rank = ValueRank(op->getResult(0)); in ComputeLayoutForward() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util_gpu.cu.cc | 31 const Tindex nnz, const Tindex input_rank, const Tindex output_rank, in ReshapeSparseTensorKernel()
64 const int64 output_rank = output_shape.dims(); in operator ()() local
|
| D | reshape_util.cc | 49 const int64_t output_rank = output_shape.dims(); in operator ()() local
103 const int64_t output_rank = target_shape_in.NumElements(); in ReshapeSparseTensor() local
|
| D | deserialize_sparse_variant_op.cc | 155 const int output_rank = output_shape->NumElements(); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 256 int64_t output_rank) { in PermuteBatchAndOffsetDims()
357 int64_t output_rank = output_shape.dimensions_size(); in ExpandInstruction() local
|
| D | gather_simplifier.cc | 113 auto output_rank = static_cast<int64_t>(start_indices_dims.size() + in ExpandInstruction() local
|
| D | instruction_fusion.cc | 212 int64_t output_rank = 0; in EffectivelyAtMostUnary() local
|
| D | hlo_evaluator.cc | 2153 const int64_t output_rank = output_shape.rank(); in CheckParameters() local
2198 int64_t output_rank = output_shape.dimensions_size(); in IterationSpaceForOutputBatchIndices() local
2215 int64_t output_rank, absl::Span<const int64_t> slice_sizes, in IterationSpaceForOutputOffsetIndices()
|
| D | shape_inference.cc | 3005 const int64_t output_rank = output_shape.rank(); in InferBroadcastShape() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | set_ops.cc | 53 DimensionHandle output_rank; in __anon1e42d5040102() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | bincount_op.cc | 48 auto output_rank = output_shape_param.rank(); in Compile() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather_nd.cc | 93 const int output_rank = indices_rank + params_rank - indices_nd - 1; in Prepare() local
|
| D | embedding_lookup_sparse.cc | 170 const int output_rank = (lookup_rank - 1) + (embedding_rank - 1); in Eval() local
|
| D | unsorted_segment.cc | 78 const int output_rank = data_rank - segment_ids_rank + 1; in ResizeOutputTensor() local
|
| D | batch_matmul.cc | 102 bool adj_x, bool adj_y, int output_rank, in ResizeOutputTensor()
362 const int output_rank = std::max(lhs_rank, rhs_rank); in Prepare() local
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
| D | unsorted_segment.cc | 110 const int output_rank = data_rank - segment_ids_rank + 1; in ResizeOutputTensor() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 384 int64_t output_rank = shape.rank(); in SourceIndexOfBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| D | einsum.cc | 527 int output_rank = desc_a.b + desc_a.f + desc_b.f; in ShuffleEinsumOutput() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 413 const int64_t output_rank = output_config.size(); in Einsum() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 3010 int output_rank = result_type.getRank(); in CanonicalizeOffset() local
3274 int64_t output_rank = output.getType().cast<ShapedType>().getRank(); in IsTFStyleBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1562 int64_t output_rank = output_shape.rank(); in AlignLayouts() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 746 const int64_t output_rank = output_type.getRank(); in VerifyConcatenationOpTypes() local
1444 auto output_rank = outermost_dim + updates_type.getRank() - outer_dims; in verify() local
|