Searched defs:tensor_in_shape (Results 1 – 6 of 6) sorted by relevance
277 const Tensor& tensor_in_shape = context->input(0); in Compute() local439 const Tensor& tensor_in_shape = context->input(0); in Compute() local526 const Tensor& tensor_in_shape = context->input(0); in Compute() local
122 const TensorShape& tensor_in_shape) { in PoolParameters()446 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
138 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
52 const TensorShape& tensor_in_shape) { in Pool3dParameters()490 const Tensor& tensor_in_shape = context->input(0); in Compute() local
229 const TensorShape& tensor_in_shape) { in Init()
315 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local535 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local