Searched refs:DimensionVector (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 94 DimensionVector slice_starts(all_to_all->shape().rank(), 0); in ExpandInstruction()
95 DimensionVector slice_strides(all_to_all->shape().rank(), 1); in ExpandInstruction()
96 DimensionVector slice_limits(all_to_all->shape().dimensions().begin(), in ExpandInstruction()
|
| 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 | 588 DimensionVector kernel_limits; in HandleConvolution()
589 DimensionVector output_limits; in HandleConvolution()
590 DimensionVector input_limits; in HandleConvolution()
619 DimensionVector valid_position_counts; in HandleConvolution()
|
| D | hlo_evaluator_typed_visitor.h | 1116 DimensionVector lhs_dim_multipliers = MakeDimMultipliers(lhs_shape);
1117 DimensionVector rhs_dim_multipliers = MakeDimMultipliers(rhs_shape);
1168 DimensionVector rhs_spatial_index(dnums.kernel_spatial_dimensions_size(),
1409 DimensionVector lhs_index(lhs_rank);
1410 DimensionVector rhs_index(rhs_rank);
1872 DimensionVector source_index(rank, 0);
2450 DimensionVector operand_index(rank);
2825 static DimensionVector MakeDimMultipliers(const Shape& shape) {
2826 DimensionVector v(shape.rank());
2844 DimensionVector window_index(rank);
|
| D | hlo_evaluator.cc | 498 DimensionVector concat_dimensions(reference_shape.dimensions().begin(), in HandleConcatenate()
512 DimensionVector source_indices(rank, 0); in HandleConcatenate()
513 DimensionVector dest_indices(concat_dimensions.size(), 0); in HandleConcatenate()
|
| D | algebraic_simplifier.cc | 5293 DimensionVector reverse_dimensions; in SwapConvOperands()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | util.h | 63 using DimensionVector = absl::InlinedVector<int64, kInlineRank>; variable
480 DimensionVector transformed_from_dimensions;
481 DimensionVector untransformed_from_dimensions;
482 DimensionVector to_dimensions;
|
| D | literal.h | 743 DimensionVector base;
744 DimensionVector step;
1073 DimensionVector minor_scan_indexes(rank, 0); in PopulateInternal()
1123 DimensionVector bounds = {times}; in Replicate()
1134 DimensionVector output_indices(bounds.size(), 0); in Replicate()
|
| D | literal.cc | 298 DimensionVector src_indexes(src_base.size(), 0); in CopySliceFromInternal()
299 DimensionVector dest_indexes(dest_base.size(), 0); in CopySliceFromInternal()
892 DimensionVector new_indices(result_shape.rank()); in SliceInternal()
916 DimensionVector result_dimensions; in Slice()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 384 struct DimensionVector { struct
418 const DimensionVector &batch_dimensions() const { return batch_dimensions_; } in batch_dimensions()
419 const DimensionVector &contracting_dimensions() const { in contracting_dimensions()
424 const DimensionVector &out_dimensions() const { return out_dimensions_; } in out_dimensions()
442 DimensionVector batch_dimensions_;
443 DimensionVector contracting_dimensions_;
446 DimensionVector out_dimensions_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emission_utils.cc | 218 DimensionVector dims_to_keep; in GetReductionKindAndContiguousComponentsImpl()
|
| D | ir_emitter_unnested.cc | 2287 DimensionVector window_size; in EmitSelectAndScatterFromMlir()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1070 DimensionVector result_dims; in CollapseFirstNDims()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2899 DimensionVector sizes; in AllToAllArray()
|