Searched defs:tensor_in_shape (Results 1 – 8 of 8) sorted by relevance
50 const TensorShape& tensor_in_shape) { in PoolParameters()290 const TensorShape& tensor_in_shape, bool propagate_nans) { in Compute()
243 const Tensor& tensor_in_shape = context->input(0); in Compute() local402 const Tensor& tensor_in_shape = context->input(0); in Compute() local482 const Tensor& tensor_in_shape = context->input(0); in Compute() local
245 const Tensor& tensor_in_shape = MklGetInput(context, 0); in Compute() local353 const Tensor& tensor_in_shape = MklGetInput(context, 0); in MklCreateLayoutsAndPrimitives() local673 const Tensor& tensor_in_shape, in SanityCheckInputs()
228 const TensorShape& tensor_in_shape) { in Init()
123 const Tensor& out_backprop, const TensorShape& tensor_in_shape, in Compute()
57 const TensorShape& tensor_in_shape) { in Pool3dParameters()494 const Tensor& tensor_in_shape = context->input(0); in Compute() local
326 const Tensor& out_backprop, const TensorShape& tensor_in_shape) { in MaxPoolingBackwardCustomKernel()
308 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local515 const TensorShape tensor_in_shape = ctx->InputShape(0); in Compile() local