Searched defs:scale_tensor (Results 1 – 4 of 4) sorted by relevance
118 Tensor scale_tensor(DataTypeToEnum<T2>::v(), {depth}); in FixedPointBatchNorm() local
297 TfLiteTensor scale_tensor = context->tensors[node->inputs->data[kScaleIndex]]; in Eval() local
686 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local1072 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local
2228 const TensorProto& scale_tensor = scale->attr().at("value").tensor(); in TrySimplify() local