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()
|