| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reshape_test_common.h | 49 std::initializer_list<int> shape_data, in ReshapeOpModel() argument
53 this->BuildWithTensorShape(input_shape, shape_shape, shape_data); in ReshapeOpModel()
57 shape_data); in ReshapeOpModel()
62 this->BuildWithHardcodedShape(input_shape, shape_data); in ReshapeOpModel()
82 std::initializer_list<int> shape_data) { in BuildWithHardcodedShape() argument
89 this->builder_.template CreateVector<int>(shape_data)) in BuildWithHardcodedShape()
96 std::initializer_list<int> shape_data) { in BuildWithTensorShape() argument
105 this->builder_.template CreateVector<int>(shape_data)) in BuildWithTensorShape()
109 if (shape_data.size() != 0) { in BuildWithTensorShape()
110 this->template PopulateTensor<int32_t>(shape_input_tensor, shape_data); in BuildWithTensorShape()
[all …]
|
| D | scatter_nd.cc | 41 const auto* shape_data = GetTensorData<IndicesT>(shape); in ResizeOutputTensor() local
44 output_shape->data[i] = shape_data[i]; in ResizeOutputTensor()
53 const IndicesT* shape_data) { in CheckShapes() argument
68 shape_data[ix + i]); in CheckShapes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 373 const Tensor* shape_data = c->input_tensor(0); in __anon45ba5f3d0602() local
374 if (shape_data == nullptr) { in __anon45ba5f3d0602()
379 dims.reserve(shape_data->NumElements()); in __anon45ba5f3d0602()
380 for (int i = 0; i < shape_data->NumElements(); ++i) { in __anon45ba5f3d0602()
381 dims.emplace_back(c->MakeDim(shape_data->flat<int32>()(i))); in __anon45ba5f3d0602()
392 const Tensor* shape_data = c->input_tensor(0); in __anon45ba5f3d0702() local
393 if (shape_data == nullptr) { in __anon45ba5f3d0702()
398 dims.reserve(shape_data->NumElements()); in __anon45ba5f3d0702()
399 for (int i = 0; i < shape_data->NumElements(); ++i) { in __anon45ba5f3d0702()
400 dims.emplace_back(c->MakeDim(shape_data->flat<int64>()(i))); in __anon45ba5f3d0702()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | reshape_test.cc | 116 const int* shape_dims_data, const int32_t* shape_data, in TestReshape() argument
125 TfLiteTensor shape_tensor = CreateTensor(shape_data, shape_dims); in TestReshape()
135 const int* shape_dims_data, const int32_t* shape_data, in TestReshapeQuantized() argument
147 TfLiteTensor shape_tensor = CreateTensor(shape_data, shape_dims); in TestReshapeQuantized()
345 const int32_t shape_data[] = {0}; in TF_LITE_MICRO_TEST() local
346 auto shape_tensor = tflite::testing::CreateTensor(shape_data, shape_dims); in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | einsum_dense.py | 142 shape_data = _analyze_einsum_string(self.equation,
146 kernel_shape, bias_shape, self.full_output_shape = shape_data
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 210 std::vector<int64_t> shape_data = {1, 1, 1, 1}; in ExpandTo4DForConvImpl() local
213 shape_data[3] = vector_length; in ExpandTo4DForConvImpl()
215 shape_data[0] = vector_length; in ExpandTo4DForConvImpl()
217 RankedTensorType::get(shape_data, elements.getType().getElementType()); in ExpandTo4DForConvImpl()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 428 std::vector<int32> shape_data = in ProcessTensorFlowReshapeOperator() local
434 for (size_t i = 0; i < shape_data.size(); i++) { in ProcessTensorFlowReshapeOperator()
435 if (shape_data[i] == -1) { in ProcessTensorFlowReshapeOperator()
440 product_non_wildcard_dims *= shape_data[i]; in ProcessTensorFlowReshapeOperator()
450 shape_data[wildcard_index] = input_flat_size / product_non_wildcard_dims; in ProcessTensorFlowReshapeOperator()
453 if (shape_data.size() == 1 && shape_data[0] == 0) { in ProcessTensorFlowReshapeOperator()
455 shape_data.clear(); in ProcessTensorFlowReshapeOperator()
459 *output_shape.mutable_dims() = shape_data; in ProcessTensorFlowReshapeOperator()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 840 const std::vector<int32> shape_data = { in ConvertSoftmaxOperator() local
842 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph); in ConvertSoftmaxOperator()
882 const std::vector<int32> shape_data = { in ConvertLogSoftmaxOperator() local
884 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph); in ConvertLogSoftmaxOperator()
1065 const auto& shape_data = shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ConvertTensorFlowReshapeOperator() local
1066 CreateReshapeShapeTensorConst(src_op.inputs[1], shape_data, tensorflow_graph); in ConvertTensorFlowReshapeOperator()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1328 SmallVector<int64_t, 4> shape_data; in fold() local
1330 shape_data.push_back(it.getSExtValue()); in fold()
1333 RankedTensorType::get(shape_data, input_type.getElementType()); in fold()
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | xnnpack_delegate.cc | 2642 const int32_t* shape_data = in VisitResizeBilinearNode() local
2645 const int32_t dim = shape_data[i]; in VisitResizeBilinearNode()
2662 subgraph, static_cast<size_t>(shape_data[0]), in VisitResizeBilinearNode()
2663 static_cast<size_t>(shape_data[1]), in VisitResizeBilinearNode()
|