| /external/executorch/backends/arm/_passes/ |
| D | convert_split_to_slice.py | 63 slice_node = create_node(
68 slice_node.meta = split_node.meta.copy()
69 slice_node.meta["val"] = slice_node.meta["val"][index]
70 output_node.replace_all_uses_with(slice_node)
|
| D | size_adjust_conv2d_pass.py | 113 slice_node = graph.create_node(
119 graph_module.graph, slice_node, q_params
123 last_node = slice_node
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/ |
| D | conv_pointwise.cc | 158 NodeContext mean_node, mul_node, slice_node; in GetOffset() local
167 IsSliceNode(graph, graph.FindProducer(slice_output_id), &slice_node)); in GetOffset()
169 absl::any_cast<SliceAttributes>(slice_node.node->operation.attributes); in GetOffset()
174 consumed_nodes->insert(slice_node.node->id); in GetOffset()
213 NodeContext slice_node; in TryFusedPointwiseConv() local
215 IsSliceNode(graph, graph.GetNode(first_node_id), &slice_node)); in TryFusedPointwiseConv()
216 const auto& first_commom_input = slice_node.inputs[0]; in TryFusedPointwiseConv()
217 auto slice_consumers = graph.FindConsumers(slice_node.outputs[0]->id); in TryFusedPointwiseConv()
224 mul_node.inputs[0]->id == slice_node.outputs[0]->id ? mul_node.inputs[1] in TryFusedPointwiseConv()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | peephole_list_idioms.cpp | 195 bool optimizeSlice(Node* slice_node, Node* list_construct_node) { in optimizeSlice()
196 auto start_val = toIValue(slice_node->input(1)); in optimizeSlice()
197 auto end_val = toIValue(slice_node->input(2)); in optimizeSlice()
198 auto step_val = toIValue(slice_node->input(3)); in optimizeSlice()
215 WithInsertPoint guard(slice_node); in optimizeSlice()
218 slice_list_construct->output()->setType(slice_node->output()->type()); in optimizeSlice()
224 slice_node->output()->replaceAllUsesWith(slice_list_construct->output()); in optimizeSlice()
225 if (mutated_lists_.count(slice_node->output())) { in optimizeSlice()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 596 Node* slice_node, InferenceContext* ctx, ShapeHandle* result, in PartialStridedSliceShape() argument
609 GetNodeAttr(slice_node->attrs(), "begin_mask", &begin_mask)); in PartialStridedSliceShape()
610 TF_RETURN_IF_ERROR(GetNodeAttr(slice_node->attrs(), "end_mask", &end_mask)); in PartialStridedSliceShape()
612 GetNodeAttr(slice_node->attrs(), "ellipsis_mask", &ellipsis_mask)); in PartialStridedSliceShape()
614 GetNodeAttr(slice_node->attrs(), "new_axis_mask", &new_axis_mask)); in PartialStridedSliceShape()
616 GetNodeAttr(slice_node->attrs(), "shrink_axis_mask", &shrink_axis_mask)); in PartialStridedSliceShape()
632 TF_RETURN_IF_ERROR(EvaluateConstantIntScalarEdge(slice_node, 1, &evaluated, in PartialStridedSliceShape()
644 TF_RETURN_IF_ERROR(EvaluateConstantIntScalarEdge(slice_node, 2, &evaluated, in PartialStridedSliceShape()
653 TF_RETURN_IF_ERROR(EvaluateConstantIntScalarEdge(slice_node, 3, &evaluated, in PartialStridedSliceShape()
663 ConstantPartialShape(ctx, slice_node, 0, &input, outer_context)); in PartialStridedSliceShape()
|
| D | shape_refiner.h | 278 Node* slice_node, shape_inference::InferenceContext* ctx,
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | remove_noop_test.cc | 307 Node* slice_node = graph.NewNode(); in TEST() local
318 slice_node->operation.type = ToString(OperationType::SLICE); in TEST()
323 slice_node->operation.attributes = attr; in TEST()
327 ASSERT_TRUE(graph.AddConsumer(slice_node->id, value1->id).ok()); in TEST()
329 ASSERT_TRUE(graph.SetProducer(slice_node->id, value2->id).ok()); in TEST()
334 UnorderedElementsAre(first_node, slice_node, second_node)); in TEST()
343 UnorderedElementsAre(first_node, slice_node, second_node)); in TEST()
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | split_cat.py | 1558 slice_node = graph.call_function(
1562 cat_user.replace_all_uses_with(slice_node)
1563 slice_node.meta.update(cat_user.meta)
1746 slice_node = graph.call_function(
1750 slice_node.meta["example_value"] = split_input.meta[
1753 new_cat_args.append(slice_node)
1754 new_cat_args_meta.append(slice_node.meta["example_value"])
1845 slice_node = graph.call_function(
1849 slice_node.meta["example_value"] = torch.narrow(
1856 graph, node, slice_node, cat_dim, unbind_dim, cat_shape
|
| /external/pytorch/torch/fx/passes/ |
| D | reinplace.py | 579 slice_node = gm.graph.create_node(
582 … 'call_function', torch.ops.aten.copy_.default, (slice_node, mutated_slice_node,), {})
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass_test.cc | 450 Node* slice_node = testing::FindNodeByName(result.get(), "slice"); in TEST() local
451 EXPECT_THAT(slice_node, in TEST()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1221 TF_ASSIGN_OR_RETURN(Node * slice_node, graph->AddNode(slice_def)); in CreateSliceNode()
1223 graph->AddEdge(concat_node, concat_out_index, slice_node, 0); in CreateSliceNode()
1224 graph->AddEdge(begin_node, 0, slice_node, 1); in CreateSliceNode()
1225 graph->AddEdge(size_node, 0, slice_node, 2); in CreateSliceNode()
1226 return slice_node; in CreateSliceNode()
1274 TF_ASSIGN_OR_RETURN(Node * slice_node, in CreateConcatNodesForRetval()
1278 return slice_node; in CreateConcatNodesForRetval()
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset9.py | 1105 slice_node = symbolic_helper._slice_helper(
1108 return symbolic_helper._squeeze_helper(g, slice_node, [dim])
|