| /external/tensorflow/tensorflow/core/framework/ |
| D | node_properties_test.cc | 61 void ValidateNodeProperties(const NodeProperties& props, const OpDef* op_def, in ValidateNodeProperties()
81 TEST(NodeProperties, Contructors) { in TEST() argument
91 NodeProperties props_from_slices(&op_def, node_def, input_types_slice, in TEST()
97 NodeProperties props_from_vectors(&op_def, node_def, input_types, in TEST()
103 TEST(NodeProperties, CreateFromNodeDef) { in TEST() argument
111 std::shared_ptr<const NodeProperties> props; in TEST()
113 NodeProperties::CreateFromNodeDef(node_def, &op_registry, &props).ok()); in TEST()
122 std::shared_ptr<const NodeProperties> props_bad; in TEST()
124 NodeProperties::CreateFromNodeDef(node_def, &op_registry, &props_bad) in TEST()
|
| D | node_properties.h | 28 struct NodeProperties { struct
30 NodeProperties(const OpDef* op_def, NodeDef node_def, in NodeProperties() function
32 : NodeProperties(op_def, std::move(node_def), in NodeProperties()
36 NodeProperties(const OpDef* _op_def, NodeDef&& _node_def, in NodeProperties() argument
51 std::shared_ptr<const NodeProperties>* props);
|
| D | node_properties.cc | 24 Status NodeProperties::CreateFromNodeDef( in CreateFromNodeDef()
26 std::shared_ptr<const NodeProperties>* props) { in CreateFromNodeDef()
33 props->reset(new NodeProperties(op_def, std::move(node_def), in CreateFromNodeDef()
|
| D | op_kernel.h | 189 const std::shared_ptr<const NodeProperties> props_;
262 const std::shared_ptr<const NodeProperties>& props,
381 std::shared_ptr<const NodeProperties> props_;
1383 const std::shared_ptr<const NodeProperties>& props, int graph_def_version,
1388 const std::shared_ptr<const NodeProperties>& props,
1394 const std::shared_ptr<const NodeProperties>& props,
|
| D | function.h | 839 const std::shared_ptr<const NodeProperties>& props,
941 const std::shared_ptr<const NodeProperties>& props) const = 0;
946 const std::shared_ptr<const NodeProperties>& props,
|
| D | op_kernel.cc | 124 : props_(std::make_shared<const NodeProperties>( in OpKernel()
231 const std::shared_ptr<const NodeProperties>& props, in OpKernelConstruction()
1580 std::shared_ptr<const NodeProperties> props; in CreateOpKernel()
1581 status->Update(NodeProperties::CreateFromNodeDef( in CreateOpKernel()
1594 const std::shared_ptr<const NodeProperties>& props, int graph_def_version, in CreateOpKernel()
1604 const std::shared_ptr<const NodeProperties>& props, in CreateOpKernel()
1614 const std::shared_ptr<const NodeProperties>& props, in CreateOpKernel()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator.h | 34 const std::shared_ptr<const NodeProperties>& props) const override;
38 const std::shared_ptr<const NodeProperties>& props,
|
| D | xla_kernel_creator.cc | 75 const std::shared_ptr<const NodeProperties>& props) const { in CanCreateKernel()
158 auto props = std::make_shared<NodeProperties>( in CreateXlaKernel()
174 const std::shared_ptr<const NodeProperties>& props, in CreateKernel()
|
| D | xla_kernel_creator_test.cc | 33 std::shared_ptr<NodeProperties> ToNodeProperties(const string& text) { in ToNodeProperties()
37 return std::make_shared<NodeProperties>(nullptr, std::move(node_def), dummy, in ToNodeProperties()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | local_executor_params.h | 28 class NodeProperties; variable
46 std::function<Status(const std::shared_ptr<const NodeProperties>&,
|
| D | function.cc | 175 Status CreateKernel(const std::shared_ptr<const NodeProperties>& props,
257 const std::shared_ptr<const NodeProperties>&, OpKernel**) { in CreateKernel() argument
344 Status CreateKernel(const std::shared_ptr<const NodeProperties>& props,
397 std::function<Status(const std::shared_ptr<const NodeProperties>&,
431 Status CreateKernel(const std::shared_ptr<const NodeProperties>& props,
483 create_kernel_ = [this](const std::shared_ptr<const NodeProperties>& props, in FunctionLibraryRuntimeImpl()
599 const std::shared_ptr<const NodeProperties>& props, OpKernel** kernel) { in CreateKernel()
604 const std::shared_ptr<const NodeProperties>& props, in CreateKernel()
659 auto new_props = std::make_shared<NodeProperties>( in CreateKernel()
937 [this, flr](const std::shared_ptr<const NodeProperties>& props, in CreateItem()
|
| D | executor.h | 237 const std::shared_ptr<const NodeProperties>& props,
|
| D | graph_runner.cc | 163 const std::shared_ptr<const NodeProperties>& props, in Run()
|
| D | kernel_benchmark_testlib.cc | 104 const std::shared_ptr<const NodeProperties>& props, in Benchmark()
|
| D | executor_test.cc | 75 [this, version](const std::shared_ptr<const NodeProperties>& props, in Create()
|
| D | executor.cc | 1365 const std::shared_ptr<const NodeProperties>& props, in CreateNonCachedKernel()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op_test.cc | 140 std::shared_ptr<const NodeProperties> props; in InitOpWithFunctionLibrary()
141 Status status = NodeProperties::CreateFromNodeDef( in InitOpWithFunctionLibrary()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 142 std::shared_ptr<const NodeProperties> props; in InitOpWithGraphVersion()
143 TF_RETURN_IF_ERROR(NodeProperties::CreateFromNodeDef( in InitOpWithGraphVersion()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph.h | 241 std::shared_ptr<NodeProperties> properties() const { return props_; } in properties()
314 void Initialize(int id, int cost_id, std::shared_ptr<NodeProperties> props,
329 std::shared_ptr<NodeProperties> props_;
730 Node* AllocateNode(std::shared_ptr<NodeProperties> props,
|
| D | graph.cc | 130 std::shared_ptr<NodeProperties> props, in Initialize()
168 props_ = std::make_shared<NodeProperties>( in UpdateProperties()
210 props_ = std::make_shared<NodeProperties>(*props_); in MaybeCopyOnWrite()
467 std::make_shared<NodeProperties>(&op_reg_data->op_def, in AddNode()
801 Node* Graph::AllocateNode(std::shared_ptr<NodeProperties> props, in AllocateNode()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 109 std::shared_ptr<const NodeProperties> props; in Init()
110 TF_RETURN_IF_ERROR(NodeProperties::CreateFromNodeDef( in Init()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_with_tf2xla.cc | 429 std::shared_ptr<const tensorflow::NodeProperties> props; in LegalizeOp()
430 tensorflow::Status status = tensorflow::NodeProperties::CreateFromNodeDef( in LegalizeOp()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_test_base.cc | 308 std::shared_ptr<const NodeProperties> props; in CreateOpKernel()
309 TF_RETURN_IF_ERROR(NodeProperties::CreateFromNodeDef( in CreateOpKernel()
452 const std::shared_ptr<const NodeProperties>& props, in RunFunction()
|
| D | single_threaded_executor_test.cc | 63 [this, version](const std::shared_ptr<const NodeProperties>& props, in Create()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 249 [handle, lib, opseg](const std::shared_ptr<const NodeProperties>& props, in InitItem()
|