Home
last modified time | relevance | path

Searched refs:split_params (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/
H A D_get_clean_triton.py106 split_params = [i.split("|") for i in launch_params_meta]
107 strip_params = [[a.strip(), b.strip()] for a, b in split_params]
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dembedding_ops_test.py559 split_params = [
562 sharded = embedding_ops.embedding_lookup(split_params, ids)
592 split_params = [
596 split_params, ids, max_norm=1.0)
628 split_params = [
632 split_params, ids, max_norm=l2_norm, transform_fn=transform)
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dconv_2d_helpers.cc188 SplitParams split_params{ in SplitWeightsForDwConv() local
213 optimized_ops::Split(split_params, weight_shape, converted_data.data(), in SplitWeightsForDwConv()
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dbuiltin_op_data_test.cc66 TfLiteSplitParams split_params; in TEST() local
/aosp_15_r20/external/ComputeLibrary/examples/
H A Dgraph_deepspeech_v0_4_1.cpp273 NodeParams split_params = { cell_name + "/split", graph.hints().target_hint }; in add_lstm_cell() local
274 …NodeID split_nid = GraphBuilder::add_split_node(graph.graph(), split_params, { graph.tail_n… in add_lstm_cell()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder.cc2260 const TfLiteSplitParams* split_params; in Parse() local
2261 RETURN_IF_ERROR(RetrieveBuiltinData(tflite_node, &split_params)); in Parse()
2262 if (split_params->num_splits == 1) { in Parse()
2302 const TfLiteSplitVParams* split_params; in Parse() local
2303 RETURN_IF_ERROR(RetrieveBuiltinData(tflite_node, &split_params)); in Parse()
2304 if (split_params->num_splits == 1) { in Parse()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dxnnpack_delegate.cc2075 const TfLiteSplitParams* split_params = in VisitNode() local
2078 node, context->tensors, split_params, in VisitNode()
2344 const TfLiteTensor* tensors, const TfLiteSplitParams* split_params, in VisitSplitNode() argument
2347 TF_LITE_ENSURE_EQ(logging_context, split_params->num_splits, num_outputs); in VisitSplitNode()