Searched defs:tensor_in_shape (Results 1 – 6 of 6) sorted by relevance
288 const Tensor& tensor_in_shape = context->input(0); in Compute() local451 const Tensor& tensor_in_shape = context->input(0); in Compute() local538 const Tensor& tensor_in_shape = context->input(0); in Compute() local
123 const TensorShape& tensor_in_shape) { in PoolParameters()445 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
52 const TensorShape& tensor_in_shape) { in Pool3dParameters()513 const Tensor& tensor_in_shape = context->input(0); in Compute() local
133 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
363 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local587 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local
233 const TensorShape& tensor_in_shape) { in Init()