Home
last modified time | relevance | path

Searched refs:shape_data (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dreshape_test.cc51 std::initializer_list<int> shape_data, in ReshapeOpModel() argument
55 BuildWithTensorShape(input_shape, shape_shape, shape_data); in ReshapeOpModel()
58 BuildWithConstantTensorShape(input_shape, shape_shape, shape_data); in ReshapeOpModel()
63 BuildWithHardcodedShape(input_shape, shape_data); in ReshapeOpModel()
81 std::initializer_list<int> shape_data) { in BuildWithHardcodedShape() argument
86 CreateReshapeOptions(builder_, builder_.CreateVector<int>(shape_data)) in BuildWithHardcodedShape()
93 std::initializer_list<int> shape_data) { in BuildWithTensorShape() argument
100 CreateReshapeOptions(builder_, builder_.CreateVector<int>(shape_data)) in BuildWithTensorShape()
103 if (shape_data.size() != 0) { in BuildWithTensorShape()
104 PopulateTensor<int32_t>(shape_input_tensor, shape_data); in BuildWithTensorShape()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner_test.cc388 const Tensor* shape_data = c->input_tensor(0); in __anon7afc939f0602() local
389 if (shape_data == nullptr) { in __anon7afc939f0602()
394 dims.reserve(shape_data->NumElements()); in __anon7afc939f0602()
395 for (int i = 0; i < shape_data->NumElements(); ++i) { in __anon7afc939f0602()
396 dims.emplace_back(c->MakeDim(shape_data->flat<int32>()(i))); in __anon7afc939f0602()
407 const Tensor* shape_data = c->input_tensor(0); in __anon7afc939f0702() local
408 if (shape_data == nullptr) { in __anon7afc939f0702()
413 dims.reserve(shape_data->NumElements()); in __anon7afc939f0702()
414 for (int i = 0; i < shape_data->NumElements(); ++i) { in __anon7afc939f0702()
415 dims.emplace_back(c->MakeDim(shape_data->flat<int64>()(i))); in __anon7afc939f0702()
[all …]
/external/u-boot/drivers/demo/
Ddemo-shape.c21 struct shape_data { struct
31 struct shape_data *data = dev_get_priv(dev); in shape_hello() argument
88 struct shape_data *data = dev_get_priv(dev); in shape_status()
96 struct shape_data *priv = dev_get_priv(dev); in set_light()
115 struct shape_data *priv = dev_get_priv(dev); in get_light()
161 struct shape_data *priv = dev_get_priv(dev); in dm_shape_probe()
177 struct shape_data *priv = dev_get_priv(dev); in dm_shape_remove()
195 .priv_auto_alloc_size = sizeof(struct shape_data),
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dsoftmax.cc94 const int32_t shape_data[4] = {1, 1, 1, input_size}; in Softmax1DQuantized() local
95 RuntimeShape shape(4, shape_data); in Softmax1DQuantized()
113 const int32_t shape_data[4] = {batch_size, 1, 1, input_size}; in Softmax2DQuantized() local
114 RuntimeShape shape(4, shape_data); in Softmax2DQuantized()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc427 std::vector<int32> shape_data = in ProcessTensorFlowReshapeOperator() local
433 for (int i = 0; i < shape_data.size(); i++) { in ProcessTensorFlowReshapeOperator()
434 if (shape_data[i] == -1) { in ProcessTensorFlowReshapeOperator()
439 product_non_wildcard_dims *= shape_data[i]; in ProcessTensorFlowReshapeOperator()
449 shape_data[wildcard_index] = input_flat_size / product_non_wildcard_dims; in ProcessTensorFlowReshapeOperator()
452 if (shape_data.size() == 1 && shape_data[0] == 0) { in ProcessTensorFlowReshapeOperator()
454 shape_data.clear(); in ProcessTensorFlowReshapeOperator()
458 *output_shape.mutable_dims() = shape_data; in ProcessTensorFlowReshapeOperator()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc834 const std::vector<int32> shape_data = { in ConvertSoftmaxOperator() local
836 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph); in ConvertSoftmaxOperator()
875 const std::vector<int32> shape_data = { in ConvertLogSoftmaxOperator() local
877 CreateReshapeShapeTensorConst(softmax_size, shape_data, tensorflow_graph); in ConvertLogSoftmaxOperator()
1057 const auto& shape_data = shape_array.GetBuffer<ArrayDataType::kInt32>().data; in ConvertTensorFlowReshapeOperator() local
1058 CreateReshapeShapeTensorConst(src_op.inputs[1], shape_data, tensorflow_graph); in ConvertTensorFlowReshapeOperator()
/external/u-boot/doc/driver-model/
DREADME.txt299 .priv_data_size = sizeof(struct shape_data),