| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | scatter_expander.cc | 183 AsInt64Slice(index->shape().dimensions())); in CheckIndexValidity()
279 AsInt64Slice(dim_numbers.inserted_window_dims()))); in ScatterLoopBody()
295 AsInt64Slice(update_slice_shape.dimensions()))); in ScatterLoopBody()
309 AsInt64Slice(operand->shape().dimensions()), in ScatterLoopBody()
310 AsInt64Slice(update_slice_with_dims_inserted->shape().dimensions()), in ScatterLoopBody()
388 updates, AsInt64Slice(dim_numbers.update_window_dims()))); in ExpandScatter()
|
| D | indexed_array_analysis.cc | 691 /*operand_shape=*/AsInt64Slice(scalar_indexed->shape().dimensions()), in FoldReshapeOfGatherNoDegenerateDims()
692 /*result_shape=*/AsInt64Slice(shape.dimensions())); in FoldReshapeOfGatherNoDegenerateDims()
728 /*operand_shape=*/AsInt64Slice( in FoldReshapeOfGatherNoDegenerateDims()
774 /*operand_shape=*/AsInt64Slice( in FoldReshapeOfGatherNoDegenerateDims()
820 AsInt64Slice(shape.dimensions())))); in ComputeArrayForReshape()
1040 AsInt64Slice(dim_numbers.lhs_contracting_dimensions()), in ComputeArrayForDotWithIndexedLhs()
1041 /*batch_dims=*/AsInt64Slice(dim_numbers.lhs_batch_dimensions()))) { in ComputeArrayForDotWithIndexedLhs()
1075 AsInt64Slice(dim_numbers.rhs_contracting_dimensions()), in ComputeArrayForDotWithIndexedRhs()
1076 /*batch_dims=*/AsInt64Slice(dim_numbers.rhs_batch_dimensions()))) { in ComputeArrayForDotWithIndexedRhs()
|
| D | gather_expander.cc | 207 AsInt64Slice(dim_numbers.collapsed_slice_dims()))); in GatherLoopBody()
368 AsInt64Slice(dim_numbers.offset_dims()), in ExpandInstruction()
|
| D | convolution_group_converter.cc | 171 S32, AsInt64Slice(expanded_filter_shape.dimensions())); in GetExpandedFilterMask()
193 PRED, AsInt64Slice(expanded_filter_shape.dimensions())); in GetExpandedFilterMask()
|
| D | triangular_solve_expander.cc | 105 auto shape_dims = AsInt64Slice(blocks_shape.dimensions()); in DiagonalBlocks()
248 return Reshape(inv_diag_blocks, AsInt64Slice(shape.dimensions())); in InvertDiagonalBlocks()
|
| D | cholesky_expander.cc | 60 auto major_dims = AsInt64Slice(a_shape.dimensions()) in CholeskyUnblocked()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 76 inline absl::Span<const int64> AsInt64Slice( in AsInt64Slice() function
82 inline absl::Span<int64> AsInt64Slice( in AsInt64Slice() function
348 return AsInt64Slice(tensor_.dimensions()).subspan(2);
352 return AsInt64Slice(tensor_.mutable_dimensions()).subspan(2);
454 return AsInt64Slice(tensor_.dimensions()).subspan(2);
459 return AsInt64Slice(tensor_.mutable_dimensions()).subspan(2);
594 return AsInt64Slice(proto_.strides());
598 return AsInt64Slice(proto_.dilations());
602 return AsInt64Slice(proto_.paddings());
606 absl::Span<int64> strides() { return AsInt64Slice(proto_.mutable_strides()); }
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 727 AsInt64Slice(shape.dimensions()); in ValidateShapeSize()
975 permutation, AsInt64Slice(shape.layout().minor_to_major()))) { in PermuteDimensions()
1055 auto common_factors = CommonFactors(AsInt64Slice(input_shape.dimensions()), in DimensionsUnmodifiedByReshape()
1056 AsInt64Slice(output_shape.dimensions())); in DimensionsUnmodifiedByReshape()
1096 AsInt64Slice(output_shape.layout().minor_to_major())), in TransposeIsBitcast()
1227 input_shape.element_type(), AsInt64Slice(input_shape.dimensions())); in ReshapeIsBitcast()
1229 output_shape.element_type(), AsInt64Slice(output_shape.dimensions())); in ReshapeIsBitcast()
1400 output_shape.element_type(), AsInt64Slice(output_shape.dimensions()), in AlignLayouts()
|
| D | layout_util.cc | 334 return AsInt64Slice(shape.layout().minor_to_major()); in MinorToMajor()
340 return AsInt64Slice(layout.minor_to_major()); in MinorToMajor()
|
| D | layout.h | 41 return Tile(AsInt64Slice(tile_proto.dimensions())); in CreateFromProto()
|
| D | util.h | 148 static inline absl::Span<const int64> AsInt64Slice( in AsInt64Slice() function
157 static inline absl::Span<const int64> AsInt64Slice( in AsInt64Slice() function
|
| D | literal_util.h | 338 AsInt64Slice(layout.minor_to_major()))); in CreateR2WithLayout()
440 AsInt64Slice(layout.minor_to_major()))); in CreateFromArrayWithLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 126 return Reshape(result, AsInt64Slice(shape.dimensions())); in StatelessRngUniformU32()
151 return Reshape(result, AsInt64Slice(shape.dimensions())); in StatelessRngUniformU64()
|
| D | sorting.cc | 31 ShapeUtil::MakeShape(S32, AsInt64Slice(input_shape.dimensions())); in TopK()
|
| D | slicing.cc | 32 auto major_dims = AsInt64Slice(shape.dimensions()) in SliceInMinorDims()
118 auto major_dims = AsInt64Slice(shape.dimensions()) in DynamicSliceInMinorDims()
|
| D | testing.cc | 53 AsInt64Slice(shape.dimensions())); in BuildFakeDataOpOnDevice()
|
| D | matrix.cc | 61 AsInt64Slice(shape.dimensions()).subspan(/*pos=*/0, /*len=*/n_dims - 2); in GetMatrixDiagonal()
100 AsInt64Slice(shape.dimensions()).subspan(/*pos=*/0, /*len=*/n_dims - 2); in TriangleMask()
|
| D | constants.cc | 29 AsInt64Slice(shape.dimensions())); in Zeros()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 33 xla::AsInt64Slice(input_shape.dimensions()); in BroadcastTo()
|
| D | util.cc | 36 xla::AsInt64Slice(shape.dimensions())); in Zeros()
|
| D | scatter.cc | 43 xla::AsInt64Slice(indices_shape.dimensions()); in XlaScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 136 CommonFactors(AsInt64Slice(input_shape.dimensions()), in SourceIndexOfReshape()
137 AsInt64Slice(output_shape.dimensions())); in SourceIndexOfReshape()
148 .Linearize(AsInt64Slice(output_shape.dimensions()) in SourceIndexOfReshape()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateful_random_ops.cc | 65 return std::make_pair(Reshape(result, xla::AsInt64Slice(shape.dimensions())), in StatefulRngUniformU32()
77 return std::make_pair(Reshape(result, xla::AsInt64Slice(shape.dimensions())), in StatefulRngUniformU64()
|
| D | conv_op_helpers.cc | 141 filter, xla::AsInt64Slice(implicit_broadcast_filter_shape.dimensions())); in ReshapeFilterForDepthwiseConvolution()
162 xla::AsInt64Slice(filter_shape.dimensions())); in ContractFilterForDepthwiseBackprop()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_rewriter.cc | 447 AsInt64Slice(dnums.kernel_spatial_dimensions()))); in MatchBackwardInput()
450 AsInt64Slice(dnums.kernel_spatial_dimensions()))); in MatchBackwardInput()
|