Home
last modified time | relevance | path

Searched refs:kScaleIndex (Results 1 – 2 of 2) sorted by relevance

/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
Dlayer_norm.cc46 const int kScaleIndex = 1; variable
57 TfLiteTensor* scale = &context->tensors[node->inputs->data[kScaleIndex]]; in Resize()
297 TfLiteTensor scale_tensor = context->tensors[node->inputs->data[kScaleIndex]]; in Eval()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc680 const size_t kScaleIndex = 1; // index of scale tensor in Compute() local
686 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute()
1065 const size_t kScaleIndex = 2; // index of scale tensor in Compute() local
1072 const Tensor& scale_tensor = MklGetInput(context, kScaleIndex); in Compute()