| /external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/ |
| D | reshape_builder.cc | 41 int num_dimensions = params->num_dimensions; in PopulateShapeFromParam() local
42 if (num_dimensions == 1 && params->shape[0] == 0) { in PopulateShapeFromParam()
46 num_dimensions = 0; in PopulateShapeFromParam()
48 for (int i = 0; i < num_dimensions; ++i) { in PopulateShapeFromParam()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape.cc | 88 int num_dimensions = params->num_dimensions; in GetOutputShapeFromParam() local
89 if (num_dimensions == 1 && params->shape[0] == 0) { in GetOutputShapeFromParam()
93 num_dimensions = 0; in GetOutputShapeFromParam()
95 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in GetOutputShapeFromParam()
96 for (int i = 0; i < num_dimensions; ++i) { in GetOutputShapeFromParam()
|
| D | tile.cc | 38 int num_dimensions) { in MultiplyShapeDims() argument
41 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in MultiplyShapeDims()
42 for (int i = 0; i < num_dimensions; ++i) { in MultiplyShapeDims()
53 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
55 TF_LITE_ENSURE_EQ(context, num_dimensions, num_multipliers); in ResizeOutput()
61 num_dimensions)); in ResizeOutput()
66 num_dimensions)); in ResizeOutput()
|
| D | topk_v2.cc | 41 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
46 TF_LITE_ENSURE_MSG(context, k <= input->dims->data[num_dimensions - 1], in ResizeOutput()
49 TfLiteIntArray* output_indexes_shape = TfLiteIntArrayCreate(num_dimensions); in ResizeOutput()
50 TfLiteIntArray* output_values_shape = TfLiteIntArrayCreate(num_dimensions); in ResizeOutput()
51 for (int i = 0; i < num_dimensions - 1; ++i) { in ResizeOutput()
55 output_indexes_shape->data[num_dimensions - 1] = k; in ResizeOutput()
56 output_values_shape->data[num_dimensions - 1] = k; in ResizeOutput()
|
| D | gather.cc | 82 const int num_dimensions = in Prepare() local
84 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in Prepare()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 338 const int num_dimensions = window.dimensions_size(); in GetGpuConvParams() local
339 CHECK_LE(num_dimensions, 3) << conv->ToString(); in GetGpuConvParams()
345 const int effective_num_dimensions = std::max(2, num_dimensions); in GetGpuConvParams()
352 CHECK_EQ(num_dimensions, dnums.input_spatial_dimensions_size()) in GetGpuConvParams()
354 CHECK_EQ(num_dimensions, dnums.kernel_spatial_dimensions_size()) in GetGpuConvParams()
356 CHECK_EQ(num_dimensions, dnums.output_spatial_dimensions_size()) in GetGpuConvParams()
384 for (int dim = 0; dim < num_dimensions; ++dim) { in GetGpuConvParams()
398 for (int dim = 0; dim < num_dimensions; ++dim) { in GetGpuConvParams()
407 for (int dim = 0; dim < num_dimensions; ++dim) { in GetGpuConvParams()
426 for (int dim = 0; dim < num_dimensions; ++dim) { in GetGpuConvParams()
[all …]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | random_grad.py | 28 def add_leading_unit_dimensions(x, num_dimensions): argument
30 [array_ops.ones([num_dimensions], dtype=dtypes.int32),
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | concatenation.cc | 59 int num_dimensions = NumDimensions(input); in Prepare() local
61 if (num_dimensions > 4) { in Prepare()
66 input->name, num_dimensions); in Prepare()
|
| /external/tensorflow/tensorflow/python/kernel_tests/random/ |
| D | random_grad_test.py | 52 num_dimensions = array_ops.placeholder(dtypes.int32)
53 ret = random_grad.add_leading_unit_dimensions(x, num_dimensions)
55 ret_val = sess.run(ret, {x: np.ones([2, 2]), num_dimensions: 2})
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | array.h | 413 int64 num_dimensions() const { return sizes_.size(); }
448 CHECK_EQ(starts.size(), num_dimensions());
449 CHECK_EQ(limits.size(), num_dimensions());
473 CHECK_EQ(from.num_dimensions(), num_dimensions());
|
| D | array_test.cc | 27 EXPECT_EQ(uninit.num_dimensions(), 2); in TEST()
|
| D | literal.h | 965 CHECK_EQ(shape().rank(), values.num_dimensions()); in PopulateFromArray()
966 for (int dim = 0; dim < values.num_dimensions(); ++dim) { in PopulateFromArray()
|
| D | literal_util.h | 416 values, LayoutUtil::GetDefaultLayoutForRank(values.num_dimensions())); in CreateFromArray()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1627 ArrayRef<int64_t> output_shape, int num_dimensions, in ComputePermutation() argument
1632 assert(output_axis < num_dimensions); in ComputePermutation()
1639 const bool is_last_axis = output_axis == num_dimensions - 1; in ComputePermutation()
1643 ComputePermutation(input_tensor, perm, output_shape, num_dimensions, in ComputePermutation()
1663 const int num_dimensions = input_tensor.getType().getRank(); in fold() local
1664 assert(perm_tensor.getType().getNumElements() == num_dimensions); in fold()
1671 for (int i = 0; i < num_dimensions; ++i) { in fold()
1683 std::vector<uint64_t> input_indices(num_dimensions); in fold()
1684 ComputePermutation(input_tensor, perm, output_shape, num_dimensions, in fold()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 71 size_t num_dimensions = flat_vector->size(); in FlatBufferIntVectorToArray() local
72 if (num_dimensions > max_size_of_buffer / sizeof(int)) { in FlatBufferIntVectorToArray()
78 for (size_t i = 0; i < num_dimensions; ++i) { in FlatBufferIntVectorToArray()
515 params->num_dimensions = new_shape->size(); in ParseOpData()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding.cc | 171 CHECK_EQ(shape.dimensions_size(), tile_assignment_.num_dimensions()); in TileOffsetForDevice()
190 CHECK_EQ(shape.dimensions_size(), tile_assignment_.num_dimensions()); in TileLimitForDevice()
343 if (shape.rank() != tile_assignment_.num_dimensions()) { in ValidateNonTuple()
|
| D | hlo_evaluator.cc | 952 const int64 num_dimensions = lengths.size(); in ComputeStrides() local
955 CHECK_EQ(num_dimensions, layout.minor_to_major_size()); in ComputeStrides()
959 std::vector<int64> strides(num_dimensions + 1); in ComputeStrides()
961 for (int64 i = 0; i < num_dimensions; i++) { in ComputeStrides()
963 const int64 index = (num_dimensions - 1) - layout.minor_to_major(i); in ComputeStrides()
967 strides[num_dimensions] = stride; in ComputeStrides()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 282 int num_dimensions; member
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | curand_10_0.inc | 81 curandGenerator_t generator, unsigned int num_dimensions) {
86 return func_ptr(generator, num_dimensions);
|