Searched refs:shape_data (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | reshape_test.cc | 51 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/ |
D | shape_refiner_test.cc | 388 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/ |
D | demo-shape.c | 21 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/ |
D | softmax.cc | 94 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/ |
D | propagate_fixed_sizes.cc | 427 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/ |
D | export_tensorflow.cc | 834 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/ |
D | README.txt | 299 .priv_data_size = sizeof(struct shape_data),
|