Searched refs:slice_node (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 549 Status ShapeRefiner::PartialStridedSliceShape(Node* slice_node, in PartialStridedSliceShape() argument
563 GetNodeAttr(slice_node->attrs(), "begin_mask", &begin_mask)); in PartialStridedSliceShape()
564 TF_RETURN_IF_ERROR(GetNodeAttr(slice_node->attrs(), "end_mask", &end_mask)); in PartialStridedSliceShape()
566 GetNodeAttr(slice_node->attrs(), "ellipsis_mask", &ellipsis_mask)); in PartialStridedSliceShape()
568 GetNodeAttr(slice_node->attrs(), "new_axis_mask", &new_axis_mask)); in PartialStridedSliceShape()
570 GetNodeAttr(slice_node->attrs(), "shrink_axis_mask", &shrink_axis_mask)); in PartialStridedSliceShape()
587 EvaluateConstantIntScalarEdge(slice_node, 1, &evaluated, &begin)); in PartialStridedSliceShape()
599 EvaluateConstantIntScalarEdge(slice_node, 2, &evaluated, &end)); in PartialStridedSliceShape()
608 EvaluateConstantIntScalarEdge(slice_node, 3, &evaluated, &stride)); in PartialStridedSliceShape()
616 TF_RETURN_IF_ERROR(ConstantPartialShape(ctx, slice_node, 0, &input)); in PartialStridedSliceShape()
|
| D | shape_refiner.h | 256 Status PartialStridedSliceShape(Node* slice_node,
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 946 auto slice_node = node_map.GetNode("slice"); in TEST_F() local
947 EXPECT_EQ(slice_node->input(0), "Conv2D"); in TEST_F()
948 EXPECT_EQ(slice_node->input(1), "slice-1-LayoutOptimizer"); in TEST_F()
949 EXPECT_EQ(slice_node->input(2), "slice-2-LayoutOptimizer"); in TEST_F()
983 auto slice_node = node_map.GetNode("slice"); in TEST_F() local
984 EXPECT_EQ(slice_node->input(0), "Conv2D"); in TEST_F()
985 EXPECT_EQ(slice_node->input(1), in TEST_F()
987 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()
|