Home
last modified time | relevance | path

Searched refs:dst_shapes (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_weights_converter_test_util.cc87 std::vector<BHWC> dst_shapes; in ConvolutionWeightsConverterTest() local
89 dst_shapes.push_back(dst_tensors[i].shape); in ConvolutionWeightsConverterTest()
97 dst_shapes, dst_ptrs)); in ConvolutionWeightsConverterTest()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc417 std::vector<BHWC> dst_shapes; in UpdateParams() local
422 dst_shapes.push_back(tensor_reserver_.Get(out_id).shape); in UpdateParams()