Home
last modified time | relevance | path

Searched defs:raw_device_name (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/
H A Dmath_ops.cc39 const char* name, const char* raw_device_name) { in Mul()
69 const char* raw_device_name) { in Conj()
87 const char* name, const char* raw_device_name) { in AddV2()
111 const char* raw_device_name) { in MatMul()
130 const char* raw_device_name) { in Neg()
150 const char* raw_device_name) { in Sum()
169 const char* name, const char* raw_device_name) { in Sub()
187 const char* name, const char* raw_device_name) { in Div()
206 const char* name, const char* raw_device_name) { in DivNoNan()
248 const char* raw_device_name) { in Exp()
[all …]
H A Darray_ops.cc38 const char* raw_device_name) { in Identity()
70 const char* raw_device_name) { in IdentityN()
85 const char* raw_device_name) { in ZerosLike()
109 const char* raw_device_name) { in Shape()
156 const char* raw_device_name) { in ExpandDims()
172 const char* raw_device_name) { in OnesLike()
H A Dnn_ops.cc47 const char* raw_device_name) { in SparseSoftmaxCrossEntropyWithLogits()
69 const char* raw_device_name) { in ReluGrad()
89 const char* raw_device_name) { in Relu()
107 const char* raw_device_name) { in BiasAdd()
129 const char* name, const char* raw_device_name) { in BiasAddGrad()
H A Dresource_variable_ops.cc39 const char* raw_device_name) { in VarHandleOp()
68 const char* name, const char* raw_device_name) { in ReadVariableOp()
87 const char* name, const char* raw_device_name) { in AssignVariableOp()
108 const char* raw_device_name) { in DestroyResourceOp()
H A Dio_ops.cc55 const char* raw_device_name) { in RestoreV2()
79 const char* raw_device_name) { in SaveV2()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/experimental/
H A Dunified_api.cc119 const char* raw_device_name) { in PYBIND11_MODULE()
189 const char* raw_device_name) { in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
H A Dvariable.cc69 absl::optional<std::string> name, const char* raw_device_name, in CreateUninitialized()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/
H A Dvariable_ops.cc38 const char* raw_device_name, in CreateUninitializedResourceVariable()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Deager_operation.h50 Status Reset(const char* op, const char* raw_device_name) override { in Reset()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/tape/
H A Dtape_operation.cc39 Status TapeOperation::Reset(const char* op, const char* raw_device_name) { in Reset()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_unified_experimental_graph.cc100 Status Reset(const char* op, const char* raw_device_name) override { in Reset()
H A Dgradients.cc303 const char* raw_device_name, ForwardOperation* forward_op_) { in Reset()
H A Dc_api_experimental.cc42 const char* raw_device_name, TF_Status* status) { in TFE_OpReset()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/
H A Dop_handler_selector_test.cc176 Status Reset(const char* op, const char* raw_device_name) override { in Reset()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.cc1286 const char* raw_device_name) { in Reset()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tfe_src.cc92 const char* raw_device_name, TF_Status* status) { in GetOp()