Searched refs:scale_tensor (Results 1 – 6 of 6) sorted by relevance
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NECropResize.cpp | 90 auto scale_tensor = support::cpp14::make_unique<Tensor>(); in configure() local
93 scale_tensor->allocator()->init(scaled_result_info); in configure()
100 _scaled_results.emplace_back(std::move(scale_tensor)); in configure()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 693 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local
713 OP_REQUIRES(context, scale_tensor.dims() == 1, in Compute()
715 scale_tensor.shape().DebugString())); in Compute()
735 scale_tensor.shape(), &dst_tensor); in Compute()
798 AllocateTFOutputs(context, scale_tensor.shape(), workspace_tf_shape, in Compute()
810 AllocateTFOutputs(context, scale_tensor.shape(), workspace_tf_shape, in Compute()
825 const U* scale_tf = scale_tensor.flat<U>().data(); in Compute()
1080 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute() local
1119 OP_REQUIRES(context, scale_tensor.dims() == 1, in Compute()
1121 scale_tensor.shape().DebugString())); in Compute()
[all …]
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLCropResize.cpp | 135 auto scale_tensor = support::cpp14::make_unique<CLTensor>(); in configure() local
138 scale_tensor->allocator()->init(scaled_result_info); in configure()
139 _scaled_results.emplace_back(std::move(scale_tensor)); in configure()
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 393 TfLiteTensor scale_tensor = context->tensors[node->inputs->data[kScaleIndex]]; in Eval() local
399 scale = PodDequantize(scale_tensor, 0); in Eval()
402 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 | 2322 const TensorProto& scale_tensor = scale->attr().at("value").tensor(); in TrySimplify() local
2323 bool scale_is_a_scalar = scale_tensor.has_tensor_shape() && in TrySimplify()
2324 scale_tensor.tensor_shape().dim_size() == 0; in TrySimplify()
|