| /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()
398 updates, AsInt64Slice(dim_numbers.update_window_dims()))); in ExpandInstruction()
|
| 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 | cholesky_expander.cc | 65 auto major_dims = AsInt64Slice(a_shape.dimensions()) in CholeskyUnblocked()
70 auto matrix_dims = AsInt64Slice(a_shape.dimensions()) in CholeskyUnblocked()
|
| D | tree_reduction_rewriter.cc | 72 MakePadding(AsInt64Slice(input_shape.dimensions()), window_dimensions, in HandleReduce()
|
| D | gather_expander.cc | 207 AsInt64Slice(dim_numbers.collapsed_slice_dims()))); in GatherLoopBody()
377 AsInt64Slice(dim_numbers.offset_dims()), in ExpandInstruction()
|
| D | batch_dot_simplification.cc | 39 !is_iota(AsInt64Slice( in ElideDegenerateBatchDimensionFromBatchDot()
|
| D | convolution_group_converter.cc | 170 S32, AsInt64Slice(expanded_filter_shape.dimensions())); in GetExpandedFilterMask()
192 PRED, AsInt64Slice(expanded_filter_shape.dimensions())); in GetExpandedFilterMask()
|
| D | triangular_solve_expander.cc | 115 auto shape_dims = AsInt64Slice(blocks_shape.dimensions()); in DiagonalBlocks()
368 return Reshape(inv_diag_blocks, AsInt64Slice(shape.dimensions())); in InvertDiagonalBlocks()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 49 ShapeUtil::MakeShape(S32, AsInt64Slice(input_shape.dimensions())); in TopK()
91 ShapeUtil::MakeShape(S32, AsInt64Slice(input_shape.dimensions())); in TopKWithPartitions()
|
| D | testing.cc | 53 AsInt64Slice(shape.dimensions())); in BuildFakeDataOpOnDevice()
|
| D | comparators.cc | 95 AsInt64Slice(shape.dimensions())); in CreateScalarComparisonComputation()
|
| D | constants.cc | 29 AsInt64Slice(shape.dimensions())); in Zeros()
|
| D | slicing.cc | 52 auto major_dims = AsInt64Slice(shape.dimensions()) in SliceInMinorDims()
139 auto major_dims = AsInt64Slice(shape.dimensions()) in DynamicSliceInMinorDims()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 80 inline absl::Span<const int64> AsInt64Slice(const T& repeated_field) { in AsInt64Slice() function
92 inline absl::Span<int64> AsInt64Slice(T* repeated_field) { in AsInt64Slice() function
340 return AsInt64Slice(tensor_.dimensions()).subspan(2); in spatial_size()
344 return AsInt64Slice(tensor_.mutable_dimensions()).subspan(2); in spatial_size()
446 return AsInt64Slice(tensor_.dimensions()).subspan(2); in input_filter_dims()
451 return AsInt64Slice(tensor_.mutable_dimensions()).subspan(2); in input_filter_dims()
590 return AsInt64Slice(proto_.strides()); in strides()
594 return AsInt64Slice(proto_.dilations()); in dilations()
598 return AsInt64Slice(proto_.paddings()); in padding()
604 absl::Span<int64> strides() { return AsInt64Slice(proto_.mutable_strides()); } in strides()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 845 AsInt64Slice(shape.dimensions()); in ValidateShapeSize()
1104 inv_permutation, AsInt64Slice(shape.layout().minor_to_major()))) { in PermuteDimensions()
1184 auto common_factors = CommonFactors(AsInt64Slice(input_shape.dimensions()), in DimensionsUnmodifiedByReshape()
1185 AsInt64Slice(output_shape.dimensions())); in DimensionsUnmodifiedByReshape()
1252 AsInt64Slice(output_shape.layout().minor_to_major())), in TransposeIsBitcast()
1385 input_shape.element_type(), AsInt64Slice(input_shape.dimensions())); in ReshapeIsBitcast()
1387 output_shape.element_type(), AsInt64Slice(output_shape.dimensions())); in ReshapeIsBitcast()
1558 output_shape.element_type(), AsInt64Slice(output_shape.dimensions()), in AlignLayouts()
1710 AsInt64Slice(normalized_shape.dimensions()); in FindTranspose021()
|
| D | layout_util.cc | 323 return AsInt64Slice(shape.layout().minor_to_major()); in MinorToMajor()
329 return AsInt64Slice(layout.minor_to_major()); in MinorToMajor()
|
| D | layout.h | 41 return Tile(AsInt64Slice(tile_proto.dimensions())); in CreateFromProto()
|
| D | util.h | 177 static inline absl::Span<const int64> AsInt64Slice( in AsInt64Slice() function
186 static inline absl::Span<const int64> AsInt64Slice( in AsInt64Slice() function
|
| D | literal_util.h | 320 AsInt64Slice(layout.minor_to_major()))); in CreateR2WithLayout()
407 AsInt64Slice(layout.minor_to_major()))); in CreateFromArrayWithLayout()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | broadcast.cc | 36 xla::AsInt64Slice(input_shape.dimensions()); in BroadcastTo()
|
| D | scatter.cc | 43 xla::AsInt64Slice(indices_shape.dimensions()); in XlaScatter()
|
| D | util.cc | 36 xla::AsInt64Slice(shape.dimensions())); in Zeros()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 223 CommonFactors(AsInt64Slice(input_shape.dimensions()), in SourceIndexOfReshape()
224 AsInt64Slice(output_shape.dimensions())); in SourceIndexOfReshape()
229 AsInt64Slice(output_shape.dimensions()) in SourceIndexOfReshape()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_rewriter.cc | 529 AsInt64Slice(dnums.kernel_spatial_dimensions()))); in MatchBackwardInput()
532 AsInt64Slice(dnums.kernel_spatial_dimensions()))); in MatchBackwardInput()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 157 xla::AsInt64Slice(shape.dimensions()))); in CopyCPUTensorToDevice()
|