Searched refs:DimensionVector (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | indexed_array_analysis.cc | 458 DimensionVector new_dims; in StripDegenerateDimensions()
477 DimensionVector new_source_shape_dims; in ReshapeToRemoveDegenerateDims()
498 DimensionVector new_output_dims; in ReshapeToRemoveDegenerateDims()
530 DimensionVector new_output_dims = [&]() { in ReshapeToAddDegenerateDims()
545 DimensionVector result; in ReshapeToAddDegenerateDims()
556 DimensionVector new_result_shape_dims; in ReshapeToAddDegenerateDims()
563 DimensionVector new_source_shape_dims = new_result_shape_dims; in ReshapeToAddDegenerateDims()
622 DimensionVector degenerate_result_dims; in FoldReshapeOfGather()
|
| D | hlo_cost_analysis.cc | 453 DimensionVector kernel_limits; in HandleConvolution()
454 DimensionVector output_limits; in HandleConvolution()
455 DimensionVector input_limits; in HandleConvolution()
484 DimensionVector valid_position_counts; in HandleConvolution()
|
| D | hlo_evaluator_typed_visitor.h | 1057 DimensionVector lhs_dim_multipliers = MakeDimMultipliers(lhs_shape); in HandleConvolution()
1058 DimensionVector rhs_dim_multipliers = MakeDimMultipliers(rhs_shape); in HandleConvolution()
1106 DimensionVector rhs_spatial_index(dnums.kernel_spatial_dimensions_size(), in HandleConvolution()
1296 DimensionVector lhs_index(lhs_rank); in HandleDotSlowPath()
1297 DimensionVector rhs_index(rhs_rank); in HandleDotSlowPath()
1888 DimensionVector source_index(rank, 0); in HandleSelectAndScatter()
1995 DimensionVector window_index(window.dimensions_size()); in HandleReduceWindow()
1996 DimensionVector operand_index(operand_literal.shape().rank()); in HandleReduceWindow()
2429 DimensionVector operand_index(rank); in HandleSlice()
2809 static DimensionVector MakeDimMultipliers(const Shape& shape) { in MakeDimMultipliers()
[all …]
|
| D | hlo_evaluator.cc | 461 DimensionVector concat_dimensions(reference_shape.dimensions().begin(), in HandleConcatenate()
475 DimensionVector source_indices(rank, 0); in HandleConcatenate()
476 DimensionVector dest_indices(concat_dimensions.size(), 0); in HandleConcatenate()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.h | 721 DimensionVector base;
722 DimensionVector step;
1096 DimensionVector minor_scan_indexes(rank, 0); in PopulateInternal()
1146 DimensionVector bounds = {times}; in Replicate()
1157 DimensionVector output_indices(bounds.size(), 0); in Replicate()
|
| D | literal.cc | 238 DimensionVector src_indexes(src_base.size(), 0); in CopySliceFromInternal()
239 DimensionVector dest_indexes(dest_base.size(), 0); in CopySliceFromInternal()
717 DimensionVector new_indices(result_shape.rank()); in SliceInternal()
733 DimensionVector result_dimensions; in Slice()
|
| D | util.h | 60 using DimensionVector = absl::InlinedVector<int64, kInlineRank>; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 853 DimensionVector window_size; in HandleSelectAndScatter()
3492 DimensionVector GetDimensionsToKeepMinorToMajor( in GetDimensionsToKeepMinorToMajor()
3494 DimensionVector input_dims(input_shape.rank(), 0); in GetDimensionsToKeepMinorToMajor()
3496 DimensionVector input_dims_to_keep; in GetDimensionsToKeepMinorToMajor()
3526 DimensionVector input_dims_to_keep_minor_to_major = in GetReductionToVectorDimensions()
3534 DimensionVector input_dims(input_shape.rank(), 0); in GetReductionToVectorDimensions()
3720 DimensionVector dims_in_elem{depth, height, width}; in ComputeMappingSchemeAndReductionKind()
3721 DimensionVector req_block_sizes{block_size_z, 1, 1}; in ComputeMappingSchemeAndReductionKind()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 874 DimensionVector result_dims; in CollapseFirstNDims()
|