| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | eval_const_tensor.cc | 76 Status TryToInferTensorOutputFromShapeNode(const Node& shape_node, in TryToInferTensorOutputFromShapeNode() argument
81 if (shape_node.type_string() != "Shape") return OkStatus(); in TryToInferTensorOutputFromShapeNode()
88 Tensor t(shape_node.output_type(0), TensorShape({src_rank})); in TryToInferTensorOutputFromShapeNode()
89 if (shape_node.output_type(0) == DT_INT32) { in TryToInferTensorOutputFromShapeNode()
105 } else if (shape_node.output_type(0) == DT_INT64) { in TryToInferTensorOutputFromShapeNode()
136 const Node* shape_node = edge->src(); in TryToInferTensorOutputFromStridedSliceNode() local
138 InferenceContext* shape_c = refiner.GetContext(shape_node); in TryToInferTensorOutputFromStridedSliceNode()
142 if (stride_node == nullptr || shape_node == nullptr) return OkStatus(); in TryToInferTensorOutputFromStridedSliceNode()
144 if (shape_node->type_string() != "Shape") return OkStatus(); in TryToInferTensorOutputFromStridedSliceNode()
178 *shape_node, shape_c, &shape_output, success, /*allow_partial=*/true)); in TryToInferTensorOutputFromStridedSliceNode()
|
| D | accumulate_n_optimizer.cc | 159 Node* shape_node; in RewriteIntoTempVariable() local
163 .Finalize(g, &shape_node)); in RewriteIntoTempVariable()
170 .Input(shape_node) in RewriteIntoTempVariable()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 206 Node* shape_node = nullptr; in ReplaceBroadcast() local
210 TF_RETURN_IF_ERROR(shape_builder.Finalize(graph, &shape_node)); in ReplaceBroadcast()
211 shape_node->set_assigned_device_name_index(send_dev); in ReplaceBroadcast()
225 TF_RETURN_IF_ERROR(shape_builder.Finalize(graph, &shape_node)); in ReplaceBroadcast()
226 shape_node->set_assigned_device_name_index(recv_dev); in ReplaceBroadcast()
231 .Input(shape_node) in ReplaceBroadcast()
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | remove_inplace_ops.cpp | 116 auto shape_node = originalInputs.at(0)->node(); in ImplicitCastForBinaryInplaceOps() local
117 if ((shape_node->kind() == prim::NumToTensor) && in ImplicitCastForBinaryInplaceOps()
118 (shape_node->inputs().at(0)->node()->kind() == aten::size)) { in ImplicitCastForBinaryInplaceOps()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | reshape_builder.cc | 86 auto* shape_node = graph_builder_->AddConstNodeWithData( in PopulateSubGraph() local
90 AddInput(TensorID(shape_node->GetID(), 0)); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_test.cc | 455 auto* shape_node = graph_view.GetNode("shape"); in TEST_F() local
456 ASSERT_NE(shape_node, nullptr); in TEST_F()
457 ASSERT_EQ(shape_node->NumRegularFanins(), 1); in TEST_F()
458 VerifyRegularFaninMatch(shape_node, 0, conv2d_node->GetName(), 0); in TEST_F()
463 VerifyRegularFaninMatch(fill_node, 0, shape_node->GetName(), 0); in TEST_F()
|
| D | constant_folding.cc | 566 bool ExtractShape(const NodeDef& shape_node, const GraphProperties& properties, in ExtractShape() argument
568 if (shape_node.op() == "Shape") { in ExtractShape()
570 properties.GetInputProperties(shape_node.name()); in ExtractShape()
583 if (shape_node.attr().count("value") == 0) { in ExtractShape()
586 const TensorProto& raw_val = shape_node.attr().at("value").tensor(); in ExtractShape()
2928 NodeDef* shape_node = optimized_graph->add_node(); in SimplifyReduction() local
2930 shape_node) in SimplifyReduction()
2934 shape_node->set_device(node->device()); in SimplifyReduction()
2935 node_map_->AddNode(shape_node->name(), shape_node); in SimplifyReduction()
2937 shape_node->add_input(AsControlDependency(reduction_indices_input)); in SimplifyReduction()
[all …]
|
| D | generic_layout_optimizer_transposer_test.cc | 3092 auto* shape_node = context.graph_view->GetNode("shape"); in TEST_F() local
3093 ASSERT_NE(shape_node, nullptr); in TEST_F()
3094 TF_ASSERT_OK(shape_transposer.TransposeNode(&context, shape_node)); in TEST_F()
3161 auto* shape_node = context.graph_view->GetNode("shape"); in TEST_F() local
3162 ASSERT_NE(shape_node, nullptr); in TEST_F()
3163 TF_ASSERT_OK(shape_transposer.TransposeNode(&context, shape_node)); in TEST_F()
|
| /external/pytorch/torch/csrc/jit/passes/onnx/ |
| D | shape_type_inference.cpp | 1753 auto shape_node = n->input(0)->node(); in SpecialPostProcess() local
1754 if (shape_node->kind() == ::c10::onnx::Shape) { in SpecialPostProcess()
1756 auto orig_type = shape_node->input()->type()->cast<TensorType>(); in SpecialPostProcess()
1764 shape_node->input()->node()->kind() == in SpecialPostProcess()
1768 shape_node->input()->node()->inputs().size())}); in SpecialPostProcess()
1772 } else if (shape_node->kind() == ::c10::prim::ListConstruct) { in SpecialPostProcess()
1779 shape_node->inputs().size(), c10::ShapeSymbol::newSymbol()); in SpecialPostProcess()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 2542 TF_ASSIGN_OR_RETURN(Node * shape_node, graph->AddNode(def)); in BuildDynamicShapeNodes()
2543 dynamic_shape_nodes->push_back(shape_node); in BuildDynamicShapeNodes()
2545 shape_node->set_assigned_device_name(src->assigned_device_name()); in BuildDynamicShapeNodes()
2546 graph->AddEdge(src, src_output, shape_node, 0); in BuildDynamicShapeNodes()
2548 graph->AddControlEdge(control_input, shape_node); in BuildDynamicShapeNodes()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compiler_test.cc | 1535 Node* shape_node = graph->AddNode(shape, &status); in TEST_F() local
1537 graph->AddControlEdge(graph->source_node(), shape_node); in TEST_F()
|