| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | shape_partition.cc | 26 std::vector<int64_t> outer_dims; in Run() local
32 outer_dims.push_back(dimension); in Run()
49 static_cast<double>(target_partition_count), 1.0 / outer_dims.size()); in Run()
53 std::vector<int64_t> dimension_partition_counts(outer_dims.size()); in Run()
54 for (int64_t i = 0; i < outer_dims.size(); ++i) { in Run()
56 std::min(static_cast<int64_t>(shape_.dimensions(outer_dims[i])), in Run()
80 shape_.dimensions(outer_dims[i]) - dimension_partition_counts[i], in Run()
|
| D | ir_emitter.cc | 2018 llvm::SmallVector<int64_t> outer_dims; in HandleSlice() local
2020 outer_dims.push_back(LayoutUtil::Major(layout, i)); in HandleSlice()
2026 outer_dims.push_back(memcpy_dim); in HandleSlice()
2031 const int64_t num_outer_loops = outer_dims.size(); in HandleSlice()
2034 loops.AddLoopsForShapeOnDimensions(slice->shape(), outer_dims, "slice"); in HandleSlice()
2633 std::vector<int64_t> outer_dims(std::next(concat_dim_layout_itr), in EmitFastConcatenate() local
2643 loops.AddLoopsForShapeOnDimensions(output_shape, outer_dims, "concat"); in EmitFastConcatenate()
2650 if (!outer_dims.empty()) { in EmitFastConcatenate()
2696 if (!outer_dims.empty()) { in EmitFastConcatenate()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | scatter_nd.cc | 58 const int outer_dims = indices.DimensionsCount() - 1; in CheckShapes() local
59 for (int i = 0; i < outer_dims; ++i) { in CheckShapes()
63 const int ix = indices.Dims(outer_dims); in CheckShapes()
64 TF_LITE_ENSURE_EQ(context, updates.DimensionsCount() - outer_dims, in CheckShapes()
66 for (int i = 0; i + outer_dims < updates.DimensionsCount(); ++i) { in CheckShapes()
67 TF_LITE_ENSURE_EQ(context, updates.Dims(i + outer_dims), in CheckShapes()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scatter_nd_op.cc | 96 const int64_t outer_dims = indices.shape().dims() - 1; in Compute() local
98 for (int i = 0; i < outer_dims; ++i) { in Compute()
102 "Dimensions [0,", outer_dims, in Compute()
104 "] must match dimensions [0,", outer_dims, in Compute()
108 const int64_t ix = indices.shape().dim_size(outer_dims); in Compute()
109 OP_REQUIRES(c, updates.shape().dims() - outer_dims == shape.dims() - ix, in Compute()
113 outer_dims, ",", updates.shape().dims(), in Compute()
116 for (int i = 0; i + outer_dims < updates.shape().dims(); ++i) { in Compute()
118 c, updates.shape().dim_size(i + outer_dims) == shape.dim_size(ix + i), in Compute()
121 "] must match dimensions [", outer_dims, ",", in Compute()
[all …]
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | ReorgLayer.cpp | 53 const unsigned int outer_dims = dst.shape().total_size() / (W * H * C); in reorg_layer() local
61 for(unsigned int b = 0; b < outer_dims; ++b) in reorg_layer()
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | gather_op_test.py | 162 outer_dims = axis
167 dest_slice = ((slice(None),) * outer_dims + (dest_index,) +
169 source_slice = ((slice(None),) * outer_dims + (source_index,) +
231 outer_dims = axis
236 dest_slice = ((slice(None),) * outer_dims + (dest_index,) +
238 source_slice = ((slice(None),) * outer_dims + (source_index,) +
|
| /external/tensorflow/tensorflow/python/ops/signal/ |
| D | fft_ops.py | 99 outer_dims = _math_ops.maximum(0, input_rank - fft_rank)
100 outer_paddings = _array_ops.zeros([outer_dims], fft_length.dtype)
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 669 const int outer_dims = indices_shape.DimensionsCount() - 1; in ScatterNd() local
670 const int indices_nd = indices_shape.Dims(outer_dims); in ScatterNd()
672 for (int i = 0; i < outer_dims; ++i) { in ScatterNd()
675 for (int i = outer_dims; i < updates_dims; ++i) { in ScatterNd()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 2427 const int64_t outer_dims = c->Rank(indices_shape) - 1; in ScatterNdShapeHelper() local
2437 c->Subshape(indices_shape, 0, outer_dims, &prefix_indices)); in ScatterNdShapeHelper()
2440 c->Subshape(updates_shape, 0, outer_dims, &prefix_updates)); in ScatterNdShapeHelper()
2445 "Dimensions [0,", outer_dims, in ScatterNdShapeHelper()
2448 " must match dimensions [0,", outer_dims, in ScatterNdShapeHelper()
2457 c->Subshape(updates_shape, outer_dims, &suffix_updates)); in ScatterNdShapeHelper()
2464 outer_dims, ",", c->Rank(updates_shape), in ScatterNdShapeHelper()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 836 const auto outer_dims = output_type.getShape().take_front(axis); in ConstFoldConcatenateOpDense() local
838 outer_dims.begin(), outer_dims.end(), 1, std::multiplies<int64_t>()); in ConstFoldConcatenateOpDense()
1427 auto outer_dims = indices_type.getRank() - 1; in verify() local
1428 auto outermost_dim = indices_type.getDimSize(outer_dims); in verify()
1431 for (auto i = 0; i < outer_dims; i++) { in verify()
1444 auto output_rank = outermost_dim + updates_type.getRank() - outer_dims; in verify()
1472 auto corresponding_dim = i - outermost_dim + outer_dims; in verify()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 2733 const int64_t outer_dims = std::max(rhs_outer_dims, lhs_outer_dims); in HandleDot() local
2734 absl::c_iota(reduce_dims, outer_dims + dnums.lhs_batch_dimensions_size()); in HandleDot()
|