Searched refs:tensor_ksize (Results 1 – 2 of 2) sorted by relevance
273 const Tensor& tensor_ksize = context->input(3); in Compute() local274 auto value_ksize = tensor_ksize.flat<int32>(); in Compute()275 ksize.resize(tensor_ksize.shape().num_elements()); in Compute()397 const Tensor& tensor_ksize = context->input(3); in Compute() local398 auto value_ksize = tensor_ksize.flat<int32>(); in Compute()399 ksize.resize(tensor_ksize.shape().num_elements()); in Compute()502 const Tensor& tensor_ksize = context->input(3); in Compute() local503 auto value_ksize = tensor_ksize.flat<int32>(); in Compute()504 ksize.resize(tensor_ksize.shape().num_elements()); in Compute()703 const Tensor& tensor_ksize = context->input(3); in Compute() local[all …]
329 const Tensor& tensor_ksize = context->input(1);330 auto value_ksize = tensor_ksize.flat<int32>();331 ksize.resize(tensor_ksize.shape().num_elements());