Searched defs:size_tensor (Results 1 – 11 of 11) sorted by relevance
47 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local204 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local232 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local258 Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local317 const Tensor size_tensor = test::AsTensor<int32>({20, 30}); in TEST() local
39 const Tensor* size_tensor = c->input_tensor(size_input_idx); in SetOutputToSizedImage() local
1526 const Tensor* size_tensor = c->input_tensor(1); in __anonb22bfa860e02() local
96 const Tensor& size_tensor = ctx->input(1); in Compute() local
72 const Tensor& size_tensor = context->input(2); in SharedValidation() local
74 const Tensor& size_tensor = MklGetInput(context, kInputSizeIndex); in ValidateMklInputs() local
114 Tensor size_tensor(DT_INT32, {}); in TEST() local
659 Tensor size_tensor(DT_INT32, TensorShape({2})); in TestQuantizeResizeBilinear() local
560 Tensor size_tensor(DT_INT32, {}); in Compile() local
960 Tensor size_tensor; in TEST_F() local