Searched refs:kDiffDstIndex (Results 1 – 1 of 1) sorted by relevance
1063 const size_t kDiffDstIndex = 0; // index of diff_dst tensor in Compute() local1070 const Tensor& diff_dst_tensor = MklGetInput(context, kDiffDstIndex); in Compute()1082 GetMklShape(context, kDiffDstIndex, &dnn_shape_diff_dst, native_format); in Compute()