Searched refs:shape_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reshape_test.cc | 122 TfLiteIntArray* shape_dims = IntArrayFromInts(shape_dims_data); in TestReshape() local
125 TfLiteTensor shape_tensor = CreateTensor(shape_data, shape_dims); in TestReshape()
143 TfLiteIntArray* shape_dims = IntArrayFromInts(shape_dims_data); in TestReshapeQuantized() local
147 TfLiteTensor shape_tensor = CreateTensor(shape_data, shape_dims); in TestReshapeQuantized()
165 const int shape_dims[] = {1, 2}; in TF_LITE_MICRO_TEST() local
173 input_dims, input_data, shape_dims, shape_int32, output_dims, output_data, in TF_LITE_MICRO_TEST()
181 const int shape_dims[] = {1, 9}; in TF_LITE_MICRO_TEST() local
189 input_dims, input, shape_dims, shape_int32, output_dims, output_data, in TF_LITE_MICRO_TEST()
197 const int shape_dims[] = {1, 4}; in TF_LITE_MICRO_TEST() local
205 input_dims, input, shape_dims, shape_int32, output_dims, output_data, in TF_LITE_MICRO_TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | scatter_nd_fuzz.cc | 55 size_t shape_dims = 1 + (data[data_ix++] % kMaxShapeDims); in FuzzImpl() local
57 TensorShape({static_cast<int64>(shape_dims)})); in FuzzImpl()
60 if (data_ix + shape_dims >= size) { in FuzzImpl()
67 for (i = 0; i < shape_dims; i++) { in FuzzImpl()
110 for (i = last; i < shape_dims; i++) { in FuzzImpl()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_type.cc | 191 llvm::SmallVector<int64_t, 4> shape_dims; in ConvertToMlirTensorType() local
192 TF_RETURN_IF_ERROR(ConvertToMlirShape(shape, &shape_dims)); in ConvertToMlirTensorType()
193 return mlir::RankedTensorType::get(shape_dims, element_type); in ConvertToMlirTensorType()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 115 auto shape_dims = AsInt64Slice(blocks_shape.dimensions()); in DiagonalBlocks() local
117 std::copy(shape_dims.begin(), shape_dims.end(), last_blocks_dims.begin()); in DiagonalBlocks()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.cc | 641 auto* shape_dims = shape->mutable_dims(); in ExtendShape() local
642 shape_dims->insert(shape_dims->begin(), size_increase, 1); in ExtendShape()
652 std::vector<int>& shape_dims = *shape->mutable_dims(); in UnextendShape() local
653 shape_dims.erase(shape_dims.begin(), shape_dims.begin() + size_reduction); in UnextendShape()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 179 const std::vector<int32> shape_dims = {static_cast<int32>(vals.size())}; in MakeConstNodeDef() local
180 TF_EXPECT_OK(TensorShapeUtils::MakeShape(shape_dims, &shape)); in MakeConstNodeDef()
|