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.py | 106 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 D | embedding_ops_test.py | 559 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 D | conv_2d_helpers.cc | 188 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 D | builtin_op_data_test.cc | 66 TfLiteSplitParams split_params; in TEST() local
|
| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A D | graph_deepspeech_v0_4_1.cpp | 273 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 D | model_builder.cc | 2260 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 D | xnnpack_delegate.cc | 2075 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()
|