Searched refs:slice_node (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 575 Node* slice_node, InferenceContext* ctx, ShapeHandle* result, in PartialStridedSliceShape() argument
588 GetNodeAttr(slice_node->attrs(), "begin_mask", &begin_mask)); in PartialStridedSliceShape()
589 TF_RETURN_IF_ERROR(GetNodeAttr(slice_node->attrs(), "end_mask", &end_mask)); in PartialStridedSliceShape()
591 GetNodeAttr(slice_node->attrs(), "ellipsis_mask", &ellipsis_mask)); in PartialStridedSliceShape()
593 GetNodeAttr(slice_node->attrs(), "new_axis_mask", &new_axis_mask)); in PartialStridedSliceShape()
595 GetNodeAttr(slice_node->attrs(), "shrink_axis_mask", &shrink_axis_mask)); in PartialStridedSliceShape()
611 TF_RETURN_IF_ERROR(EvaluateConstantIntScalarEdge(slice_node, 1, &evaluated, in PartialStridedSliceShape()
623 TF_RETURN_IF_ERROR(EvaluateConstantIntScalarEdge(slice_node, 2, &evaluated, in PartialStridedSliceShape()
632 TF_RETURN_IF_ERROR(EvaluateConstantIntScalarEdge(slice_node, 3, &evaluated, in PartialStridedSliceShape()
642 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/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 966 auto slice_node = node_map.GetNode("slice"); in TEST_F() local
967 EXPECT_EQ(slice_node->input(0), "Conv2D"); in TEST_F()
968 EXPECT_EQ(slice_node->input(1), "slice-1-LayoutOptimizer"); in TEST_F()
969 EXPECT_EQ(slice_node->input(2), "slice-2-LayoutOptimizer"); in TEST_F()
1003 auto slice_node = node_map.GetNode("slice"); in TEST_F() local
1004 EXPECT_EQ(slice_node->input(0), "Conv2D"); in TEST_F()
1005 EXPECT_EQ(slice_node->input(1), in TEST_F()
1007 EXPECT_EQ(slice_node->input(2), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | increase_dynamism_for_auto_jit_pass_test.cc | 449 Node* slice_node = testing::FindNodeByName(result.get(), "slice"); in TEST() local
450 EXPECT_THAT(slice_node, in TEST()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1007 Node* slice_node = graph->AddNode(slice_def, &s); in CreateSliceNode() local
1010 graph->AddEdge(concat_node, concat_out_index, slice_node, 0); in CreateSliceNode()
1011 graph->AddEdge(begin_node, 0, slice_node, 1); in CreateSliceNode()
1012 graph->AddEdge(size_node, 0, slice_node, 2); in CreateSliceNode()
1013 return slice_node; in CreateSliceNode()
1061 TF_ASSIGN_OR_RETURN(Node * slice_node, in CreateConcatNodesForRetval()
1065 return slice_node; in CreateConcatNodesForRetval()
|