Searched refs:summand_idx (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.cc | 1696 int summand_idx = context->num_inputs() / 2 - 1 - 2; in ExtendConvFwdParams() local
1697 DataType summand_type = this->input_type(summand_idx); in ExtendConvFwdParams()
1744 int summand_idx = context->num_inputs() / 2 - 1; in AllocateOutputTensor() local
1746 summand_idx -= 2; in AllocateOutputTensor()
1747 DataType summand_type = this->input_type(summand_idx); in AllocateOutputTensor()
1751 Tensor& summand = const_cast<Tensor&>(MklGetInput(context, summand_idx)); in AllocateOutputTensor()
1753 GetMklShape(context, summand_idx, &summand_mkl_shape); in AllocateOutputTensor()
1769 context, summand_idx, 0, output_tensor, summand_mkl_shape, false), in AllocateOutputTensor()
1780 const Tensor& summand = MklGetInput(context, summand_idx); in AllocateOutputTensor()
1785 GetMklShape(context, summand_idx, &summand_mkl_shape); in AllocateOutputTensor()
|