Searched defs:scale_tensor (Results 1 – 3 of 3) sorted by relevance
118 Tensor scale_tensor(DataTypeToEnum<T2>::v(), {depth}); in FixedPointBatchNorm() local
519 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local843 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local
2124 const TensorProto& scale_tensor = scale->attr().at("value").tensor(); in TrySimplify() local