Home
last modified time | relevance | path

Searched refs:kInputTensorIndexInput (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc51 MklGetInput(context, this->kInputTensorIndexInput); in Compute()
53 GetMklShape(context, this->kInputTensorIndexInput, &dnn_shape_input, in Compute()
Dmkl_maxpooling_op.cc52 MklGetInput(context, this->kInputTensorIndexInput); in Compute()
54 GetMklShape(context, this->kInputTensorIndexInput, &dnn_shape_input, in Compute()
Dmkl_pooling_ops_common.h697 const int kInputTensorIndexInput = 0; variable