Searched refs:scale_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_fused_batch_norm_op.cc | 519 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local
539 OP_REQUIRES(context, scale_tensor.dims() == 1, in Compute()
541 scale_tensor.shape().DebugString())); in Compute()
568 HandleEmptyInput(context, tf_shape_src, scale_tensor.shape(), in Compute()
586 AllocateTFOutputs(context, scale_tensor.shape(), &batch_mean_tensor, in Compute()
623 const T* scale_tf = scale_tensor.flat<T>().data(); in Compute()
843 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local
879 OP_REQUIRES(context, scale_tensor.dims() == 1, in Compute()
881 scale_tensor.shape().DebugString())); in Compute()
896 HandleEmptyInput(context, tf_shape_src, scale_tensor.shape(), in Compute()
[all …]
|
| D | quantized_batch_norm_op.cc | 118 Tensor scale_tensor(DataTypeToEnum<T2>::v(), {depth}); in FixedPointBatchNorm() local
119 auto scale_flat = scale_tensor.flat<T2>(); in FixedPointBatchNorm()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 2124 const TensorProto& scale_tensor = scale->attr().at("value").tensor(); in TrySimplify() local
2125 bool scale_is_a_scalar = scale_tensor.has_tensor_shape() && in TrySimplify()
2126 scale_tensor.tensor_shape().dim_size() == 0; in TrySimplify()
|