Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_requantization_range_per_channel_op.cc45 const Tensor& input = ctx->input(kInputTensorIndex); in Compute()
112 const int kInputTensorIndex = 0; member in tensorflow::MklRequantizationRangePerChannelOp
Dmkl_requantize_per_channel_op.cc50 const Tensor& input = ctx->input(kInputTensorIndex); in Compute()
153 const int kInputTensorIndex = 0; member in tensorflow::MklRequantizePerChannelOp
Dmkl_slice_op.cc69 const int kInputTensorIndex = 0; in ValidateMklInputs() local
72 const Tensor& input = MklGetInput(context, kInputTensorIndex); in ValidateMklInputs()
77 GetMklShape(context, kInputTensorIndex, &input_mkl_shape); in ValidateMklInputs()