Searched refs:kInputIndexWeight (Results 1 – 3 of 3) sorted by relevance
60 const Tensor& weight_tensor = ctx->input(this->kInputIndexWeight); in Compute()66 GetMklShape(ctx, this->kInputIndexWeight, &weight_mkl_shape, native_format); in Compute()
164 MklGetInput(context, this->kInputIndexWeight); in Compute()169 GetMklShape(context, this->kInputIndexWeight, &weight_mkl_shape); in Compute()
502 const int kInputIndexWeight = 1; variable