Searched defs:scale_tensor (Results 1 – 6 of 6) sorted by relevance
90 auto scale_tensor = support::cpp14::make_unique<Tensor>(); in configure() local
135 auto scale_tensor = support::cpp14::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
693 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local1080 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local
2322 const TensorProto& scale_tensor = scale->attr().at("value").tensor(); in TrySimplify() local