Home
last modified time | relevance | path

Searched defs:k_value (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Din_topk_op.h38 int64_t k_value = -1; member
H A Din_topk_op.cc44 int64_t k_value = k_; in Compute() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DGEMMLowpMatrixMultiplyReshapedOnlyRHS.cpp116 void validate_configuration(unsigned int m_value, unsigned int n_value, unsigned int k_value, unsig… in validate_configuration()
H A DGEMMMatrixMultiplyNative.cpp249 void validate_configuration(unsigned int m_value, unsigned int n_value, unsigned int k_value, unsig… in validate_configuration()
H A DGEMMMatrixMultiplyReshapedOnlyRHS.cpp268 bool validate_configuration(unsigned int m_value, unsigned int n_value, unsigned int k_value, unsig… in validate_configuration()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc1492 const int32_t k_value = input_k.GetBuffer<ArrayDataType::kInt32>().data[0]; in ProcessTopkV2Operator() local