| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_runner.cc | 139 const int num_dimensions = window.dimensions_size(); in RunCudnnConvImpl() local
140 CHECK_LE(num_dimensions, 3); in RunCudnnConvImpl()
141 CHECK_GE(num_dimensions, 1); in RunCudnnConvImpl()
146 const int effective_num_dimensions = std::max(2, num_dimensions); in RunCudnnConvImpl()
156 CHECK_EQ(num_dimensions, dnums.input_spatial_dimensions_size()); in RunCudnnConvImpl()
157 CHECK_EQ(num_dimensions, dnums.kernel_spatial_dimensions_size()); in RunCudnnConvImpl()
158 CHECK_EQ(num_dimensions, dnums.output_spatial_dimensions_size()); in RunCudnnConvImpl()
186 for (int dim = 0; dim < num_dimensions; ++dim) { in RunCudnnConvImpl()
199 for (int dim = 0; dim < num_dimensions; ++dim) { in RunCudnnConvImpl()
208 for (int dim = 0; dim < num_dimensions; ++dim) { in RunCudnnConvImpl()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape.cc | 80 int num_dimensions = params->num_dimensions; in GetOutputShapeFromParam() local
81 if (num_dimensions == 1 && params->shape[0] == 0) { in GetOutputShapeFromParam()
85 num_dimensions = 0; in GetOutputShapeFromParam()
87 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in GetOutputShapeFromParam()
88 for (int i = 0; i < num_dimensions; ++i) { in GetOutputShapeFromParam()
|
| D | tile.cc | 36 int num_dimensions) { in MultiplyShapeDims() argument
39 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in MultiplyShapeDims()
40 for (int i = 0; i < num_dimensions; ++i) { in MultiplyShapeDims()
51 const int num_dimensions = NumDimensions(input); in ResizeOutput() local
53 TF_LITE_ENSURE_EQ(context, num_dimensions, num_multipliers); in ResizeOutput()
59 num_dimensions)); in ResizeOutput()
64 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 | 80 const int num_dimensions = in Prepare() local
82 TfLiteIntArray* output_shape = TfLiteIntArrayCreate(num_dimensions); in Prepare()
|
| /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/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 | 1031 CHECK_EQ(shape().rank(), values.num_dimensions()); in PopulateFromArray()
1032 for (int dim = 0; dim < values.num_dimensions(); ++dim) { in PopulateFromArray()
|
| D | literal_util.h | 449 values, LayoutUtil::GetDefaultLayoutForRank(values.num_dimensions())); in CreateFromArray()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 39 int num_dimensions = flat_vector->Length(); in FlatBufferIntVectorToArray() local
40 if (num_dimensions > max_size_of_buffer / sizeof(int)) { in FlatBufferIntVectorToArray()
46 for (int i = 0; i < num_dimensions; ++i) { in FlatBufferIntVectorToArray()
463 params->num_dimensions = new_shape->Length(); 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()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | builtin_op_data.h | 251 int num_dimensions; member
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | curand_10_0.inc | 77 curandSetQuasiRandomGeneratorDimensions(curandGenerator_t generator, unsigned int num_dimensions) {
81 return func_ptr(generator, num_dimensions);
|