Home
last modified time | relevance | path

Searched refs:TestReshape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/micro/kernels/
Dreshape_test.cc115 void TestReshape(const int* input_dims_data, const float* input_data, in TestReshape() function
172 tflite::testing::TestReshape( in TF_LITE_MICRO_TEST()
188 tflite::testing::TestReshape( in TF_LITE_MICRO_TEST()
204 tflite::testing::TestReshape( in TF_LITE_MICRO_TEST()
248 tflite::testing::TestReshape(input_dims, input_float, shape_dims, shape_int32, in TF_LITE_MICRO_TEST()
280 tflite::testing::TestReshape(input_dims, input_float, shape_dims, shape_int32, in TF_LITE_MICRO_TEST()
312 tflite::testing::TestReshape(input_dims, input_float, shape_dims, shape_int32, in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/core/framework/
Dtensor_test.cc401 void TestReshape(std::initializer_list<int64> sizes) { in TestReshape() function in tensorflow::__anona5e40b380111::TensorReshapeTest
407 void TestReshape(std::initializer_list<int64> sizes) { in TestReshape() function in tensorflow::__anona5e40b380111::TensorReshapeTest
446 TestReshape<TTypes<float, N>::Tensor, &Tensor::shaped<float, N>>( \ in TEST_F()
448 TestReshape<TTypes<float, N>::ConstTensor, &Tensor::shaped<float, N>>( \ in TEST_F()
450 TestReshape<TTypes<float, N>::UnalignedTensor, \ in TEST_F()
452 TestReshape<TTypes<float, N>::UnalignedConstTensor, \ in TEST_F()
454 TestReshape<TTypes<float, N>::Tensor, \ in TEST_F()
456 TestReshape<TTypes<float, N>::ConstTensor, \ in TEST_F()
458 TestReshape<TTypes<int32, N>::Tensor, \ in TEST_F()
460 TestReshape<TTypes<int32, N>::ConstTensor, \ in TEST_F()
[all …]
/external/tensorflow/tensorflow/go/
Dtensor_test.go362 func TestReshape(t *testing.T) { func