Searched defs:scale_tensor (Results 1 – 8 of 8) sorted by relevance
91 auto scale_tensor = std::make_unique<Tensor>(); in configure() local
44 const Tensor& scale_tensor = MklGetInput(ctx, kScaleIndex); in Compute() local
232 const Tensor& scale_tensor = ctx->input(2); in ExtendMklMatMulParams() local
710 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local1134 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local
133 auto scale_tensor = std::make_unique<CLTensor>(); in configure() local
118 Tensor scale_tensor(DataTypeToEnum<T2>::v(), {depth}); in FixedPointBatchNorm() local
393 TfLiteTensor scale_tensor = context->tensors[node->inputs->data[kScaleIndex]]; in Eval() local
2321 const TensorProto& scale_tensor = scale->attr().at("value").tensor(); in TrySimplify() local