Searched refs:kSrcIndex (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_dequantize_op.cc | 56 const Tensor& src_tensor = MklGetInput(ctx, kSrcIndex); in Compute()
64 GetMklShape(ctx, kSrcIndex, &src_mkl_shape); in Compute()
168 const size_t kSrcIndex = 0; member in tensorflow::MklDequantizeOp
|
| D | mkl_fused_batch_norm_op.cc | 679 const size_t kSrcIndex = 0; // index of src input tensor in Compute() local
685 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); in Compute()
693 GetMklShape(context, kSrcIndex, &dnn_shape_src, native_format); in Compute()
1064 const size_t kSrcIndex = 1; // index of src input tensor in Compute() local
1071 const Tensor& src_tensor = MklGetInput(context, kSrcIndex); in Compute()
1081 GetMklShape(context, kSrcIndex, &dnn_shape_src, native_format); in Compute()
|