Searched defs:axis_data (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | split_test.cc | 29 const int* axis_dims_data, const int32_t* axis_data, in TestSplitTwoOutputsFloat()
83 const int* axis_dims_data, const int32_t* axis_data, in TestSplitFourOutputsFloat()
157 const int* axis_dims_data, const int32_t* axis_data, in TestSplitTwoOutputsQuantized()
213 const int* axis_dims_data, const int32_t* axis_data, in TestSplitTwoOutputsQuantized32()
276 const int32_t axis_data[] = {0}; in TF_LITE_MICRO_TEST() local
296 const int32_t axis_data[] = {1}; in TF_LITE_MICRO_TEST() local
316 const int32_t axis_data[] = {2}; in TF_LITE_MICRO_TEST() local
336 const int32_t axis_data[] = {3}; in TF_LITE_MICRO_TEST() local
356 const int32_t axis_data[] = {-4}; in TF_LITE_MICRO_TEST() local
375 const int32_t axis_data[] = {0}; in TF_LITE_MICRO_TEST() local
[all …]
|
| D | reduce_test.cc | 93 const int* axis_dims_data, const int32_t* axis_data, in TestMeanFloatInput4D()
121 const int* axis_dims_data, const int32_t* axis_data, in TestReduceOpFloat()
151 const int32_t* axis_data, const int* output_dims_data, in TestReduceOpQuantized()
188 const int32_t* axis_data, const int* output_dims_data, in TestMeanOpQuantized()
485 const int32_t axis_data[] = {1, 0, -3, -3}; in TF_LITE_MICRO_TEST() local
503 const int32_t axis_data[] = {0, 2}; in TF_LITE_MICRO_TEST() local
519 const int32_t axis_data[] = {1, 1}; in TF_LITE_MICRO_TEST() local
543 const int32_t axis_data[] = {1, 1}; in TF_LITE_MICRO_TEST() local
|
| D | reduce.cc | 127 void ResolveAxis(const int* axis_data, int axis_count, in ResolveAxis()
|
| D | split_v_test.cc | 35 const int* axis_dims_data, const int32_t* axis_data, in TestSplitVFloat()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_gather_attributes.cc | 46 const auto& axis_data = indices_array.GetBuffer<ArrayDataType::kInt32>().data; in Run() local
|
| D | resolve_tensorflow_concat.cc | 64 const auto& axis_data = axis_array.GetBuffer<ArrayDataType::kInt32>().data; in Run() local
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | pooling_layer_builder.cc | 151 const int* axis_data = GetTensorData<int>(axis); in IsMeanOpSupported() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | reduce.cc | 288 void ResolveAxis(const int* axis_data, int axis_count, in ResolveAxis()
|