Searched refs:dim_map (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | test_utils.py | 37 dim_map = {d: i for i, d in enumerate(data_format_src)}
38 transpose_dims = [dim_map[d] for d in data_format_dst]
58 dim_map = {d: i for i, d in enumerate(data_format_src)}
59 permuted_dims = [dims[dim_map[d]] for d in data_format_dst]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_scatter_utils.cc | 52 MakeOperandStartIndexPermutations(absl::Span<const int64_t> dim_map, in MakeOperandStartIndexPermutations() argument
56 absl::c_copy(dim_map, std::back_inserter(perm)); in MakeOperandStartIndexPermutations()
58 if (!absl::c_linear_search(dim_map, i)) { in MakeOperandStartIndexPermutations()
|
| D | space_to_batch_converter.cc | 1080 std::vector<int64_t>& dim_map, bool check_lhs) { in CanPropagate() argument
1084 dim_map[DimMapper(SpaceToBatchDimMap::kSpace0)]) { in CanPropagate()
1090 dim_map[DimMapper(SpaceToBatchDimMap::kBatch)] || in CanPropagate()
1092 dim_map[DimMapper(SpaceToBatchDimMap::kFeature)]) { in CanPropagate()
1099 dim_map[DimMapper(SpaceToBatchDimMap::kSpace0)]) { in CanPropagate()
1105 dim_map[DimMapper(SpaceToBatchDimMap::kBatch)] || in CanPropagate()
1107 dim_map[DimMapper(SpaceToBatchDimMap::kFeature)]) { in CanPropagate()
1982 std::vector<int64_t> dim_map(NumMappedDims()); in Propagate() local
1983 dim_map[DimMapper(SpaceToBatchDimMap::kBatch)] = new_batch_dim; in Propagate()
1984 dim_map[DimMapper(SpaceToBatchDimMap::kFeature)] = new_feature_dim; in Propagate()
[all …]
|
| D | algebraic_simplifier.cc | 1266 auto dim_map = ComputeBitcastDimMap(bitcast->shape(), copy->shape()); in SwapCopyBitcastCopy() local
1267 if (!dim_map.has_value()) { in SwapCopyBitcastCopy()
1271 std::vector<std::vector<int64_t>> operand_map = dim_map.value(); in SwapCopyBitcastCopy()
|
| /external/angle/scripts/ |
| D | update_extension_data.py | 193 dim_map = {
204 if key in dim_map:
206 mapped_values = dim_map[key]
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_io.cpp | 187 DSizes<int, NumDims> dim_map; in test_block_io_copy_using_reordered_dimensions() local
189 dim_map[j] = static_cast<int>(output_to_input_dim_map[j]); in test_block_io_copy_using_reordered_dimensions()
190 TensorBlockIO::Copy(dst, src, /*dst_to_src_dim_map=*/dim_map); in test_block_io_copy_using_reordered_dimensions()
205 DSizes<int, NumDims> dim_map; in test_block_io_copy_using_reordered_dimensions() local
207 dim_map[j] = static_cast<int>(input_to_output_dim_map[j]); in test_block_io_copy_using_reordered_dimensions()
208 TensorBlockIO::Copy(dst, src, /*dst_to_src_dim_map=*/dim_map); in test_block_io_copy_using_reordered_dimensions()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorBlock.h | 1197 const DimensionsMap& dim_map = dst_to_src_dim_map;
1200 int num_squeezable_dims = NumSqueezableInnerDims(dim_map);
1232 NumDims == 0 ? 1 : dim_map[dst_stride1_dim];
1242 const IndexType src_stride = src.strides[dim_map[dst_dim]];
1268 it[idx].input_stride = src.strides[dim_map[dst_dim]];
1349 static int NumSqueezableInnerDims(const DimensionsMap& dim_map) {
1353 if (dim_map[dim] != dim) break;
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1519 absl::InlinedVector<int64_t, 8> dim_map; in AlignLayouts() local
1520 dim_map.reserve(simple_output_shape->rank()); in AlignLayouts()
1523 dim_map.push_back(i); in AlignLayouts()
1527 d = dim_map[d]; in AlignLayouts()
|