Searched refs:scale_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 686 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local
706 OP_REQUIRES(context, scale_tensor.dims() == 1, in Compute()
708 scale_tensor.shape().DebugString())); in Compute()
728 scale_tensor.shape(), &dst_tensor); in Compute()
791 AllocateTFOutputs(context, scale_tensor.shape(), workspace_tf_shape, in Compute()
803 AllocateTFOutputs(context, scale_tensor.shape(), workspace_tf_shape, in Compute()
818 const U* scale_tf = scale_tensor.flat<U>().data(); in Compute()
1072 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local
1111 OP_REQUIRES(context, scale_tensor.dims() == 1, in Compute()
1113 scale_tensor.shape().DebugString())); in Compute()
[all …]
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 297 TfLiteTensor scale_tensor = context->tensors[node->inputs->data[kScaleIndex]]; in Eval() local
303 scale = PodDequantize(scale_tensor, 0); in Eval()
306 scale = scale_tensor.data.f[0]; in Eval()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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 | 2228 const TensorProto& scale_tensor = scale->attr().at("value").tensor(); in TrySimplify() local
2229 bool scale_is_a_scalar = scale_tensor.has_tensor_shape() && in TrySimplify()
2230 scale_tensor.tensor_shape().dim_size() == 0; in TrySimplify()
|