| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 45 const int64 output_rank = output_shape.dims(); in operator ()() local
99 const int64 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 | 254 int64 output_rank) { in PermuteBatchAndOffsetDims()
331 int64 output_rank = output_shape.dimensions_size(); in ExpandInstruction() local
|
| D | instruction_fusion.cc | 202 int64 output_rank = 0; in EffectivelyAtMostUnary() local
|
| D | hlo_evaluator.cc | 1303 const int64 output_rank = output_shape.rank(); in CheckParameters() local
1452 int64 output_rank = output_shape.dimensions_size(); in IterationSpaceForOutputBatchIndices() local
1469 int64 output_rank, absl::Span<const int64> slice_sizes, in IterationSpaceForOutputOffsetIndices()
|
| D | dynamic_dimension_inference.cc | 1223 int64 output_rank = hlo->shape().rank(); in HandleGather() local
|
| D | shape_inference.cc | 2865 const int64 output_rank = output_shape.rank(); in InferBroadcastShape() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | set_ops.cc | 53 DimensionHandle output_rank; in __anonda6063e60102() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | gather_nd.cc | 92 const int output_rank = indices_rank + params_rank - indices_nd - 1; in Prepare() local
|
| D | embedding_lookup_sparse.cc | 168 const int output_rank = (lookup_rank - 1) + (embedding_rank - 1); in Eval() local
|
| D | batch_matmul.cc | 102 bool adj_x, bool adj_y, int output_rank, in ResizeOutputTensor()
358 const int output_rank = std::max(lhs_rank, rhs_rank); in Prepare() local
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | multi_head_attention.py | 122 def _get_output_shape(output_rank, known_last_dims): argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 355 int64 output_rank = shape.rank(); in SourceIndexOfBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 380 const int64 output_rank = output_config.size(); in Einsum() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 1195 int64_t output_rank = output.getType().cast<ShapedType>().getRank(); in IsTFStyleBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 597 const int64_t output_rank = output_type.getRank(); in VerifyConcatenationOpTypes() local
1062 auto output_rank = outermost_dim + updates_type.getRank() - outer_dims; in Verify() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1461 int64 output_rank = output_shape.rank(); in AlignLayouts() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 341 int64 output_rank) { in ComputeDimensionIndexMapping()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 3149 int64_t output_rank = max_rank_ty.getRank(); in inferReturnTypes() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 765 const int output_rank = in MaterializeReductionIndices() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2144 const int64 output_rank = output_shape.rank(); in Map() local
|