Home
last modified time | relevance | path

Searched refs:TFE_Op (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/c/eager/
Dc_api.h246 typedef struct TFE_Op TFE_Op; typedef
248 TF_CAPI_EXPORT extern TFE_Op* TFE_NewOp(TFE_Context* ctx,
251 TF_CAPI_EXPORT extern void TFE_DeleteOp(TFE_Op* op);
256 TF_CAPI_EXPORT extern const char* TFE_OpGetName(const TFE_Op* op,
258 TF_CAPI_EXPORT extern TFE_Context* TFE_OpGetContext(const TFE_Op* op,
261 TF_CAPI_EXPORT extern void TFE_OpSetDevice(TFE_Op* op, const char* device_name,
264 TF_CAPI_EXPORT extern const char* TFE_OpGetDevice(const TFE_Op* op,
267 TF_CAPI_EXPORT extern void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input,
270 TF_CAPI_EXPORT extern void TFE_OpAddInputList(TFE_Op* op,
284 TF_CAPI_EXPORT extern int TFE_OpGetFlatInputCount(const TFE_Op* op,
[all …]
Dtfe_op_internal.h27 typedef struct TFE_Op TFE_Op; typedef
31 DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionOperation, TFE_Op);
32 DEFINE_CONVERSION_FUNCTIONS(tensorflow::ImmediateExecutionOperation*, TFE_Op*);
Dc_api.cc636 TFE_Op* TFE_NewOp(TFE_Context* ctx, const char* op_or_function_name, in TFE_NewOp()
648 void TFE_DeleteOp(TFE_Op* op) { in TFE_DeleteOp()
656 const char* TFE_OpGetName(const TFE_Op* op, TF_Status* status) { in TFE_OpGetName()
660 TFE_Context* TFE_OpGetContext(const TFE_Op* op, TF_Status* status) { in TFE_OpGetContext()
664 void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) { in TFE_OpSetDevice()
668 const char* TFE_OpGetDevice(const TFE_Op* op, TF_Status* status) { in TFE_OpGetDevice()
672 void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) { in TFE_OpAddInput()
676 void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs, in TFE_OpAddInputList()
684 extern int TFE_OpGetFlatInputCount(const TFE_Op* op, TF_Status* status) { in TFE_OpGetFlatInputCount()
688 extern TFE_TensorHandle* TFE_OpGetFlatInput(const TFE_Op* op, int index, in TFE_OpGetFlatInput()
[all …]
Dc_api_test_util.h71 TFE_Op* AddOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
74 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
77 TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a);
80 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a);
86 TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input,
Dtfe_op_attrs_internal.h28 typedef struct TFE_Op TFE_Op; typedef
34 void SetOpAttrValueScalar(TFE_Context* ctx, TFE_Op* op,
Dc_api_test_util.cc199 TFE_Op* op = TFE_NewOp(ctx, "VarHandleOp", status); in TestVariable()
247 TFE_Op* AddOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b) { in AddOp()
250 TFE_Op* op = TFE_NewOp(ctx, "AddV2", status); in AddOp()
262 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b) { in MatMulOp()
265 TFE_Op* op = TFE_NewOp(ctx, "MatMul", status); in MatMulOp()
277 TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a) { in IdentityOp()
280 TFE_Op* op = TFE_NewOp(ctx, "Identity", status); in IdentityOp()
290 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a) { in ShapeOp()
293 TFE_Op* op = TFE_NewOp(ctx, "Shape", status); in ShapeOp()
317 TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input, in MinOp()
[all …]
Dc_api_test.cc63 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_InitOp()
85 TFE_Op* matmul = TFE_NewOp(ctx, "MatMul", status); in BM_Execute()
127 TFE_Op* identity = TFE_NewOp(ctx, "Identity", status); in BM_Execute_Identity()
430 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopy()
486 TFE_Op* matmul = MatMulOp(ctx, m, m); in SetAndGetOpDevices()
573 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST()
634 TFE_Op* add_op = AddOp(ctx, n, m); in ExecuteAdd()
728 TFE_Op* matmul = MatMulOp(ctx, m, m); in Execute_MatMul_CPU()
765 TFE_Op* matmul = MatMulOp(ctx, m1, m2); in Execute_MatMul_CPU_Runtime_Error()
769 TFE_Op* matmul2 = MatMulOp(ctx, m1, m1); in Execute_MatMul_CPU_Runtime_Error()
[all …]
Dc_api_experimental.h32 TF_CAPI_EXPORT extern void TFE_OpReset(TFE_Op* op_to_reset,
297 TFE_Op* op, TFE_CancellationManager* cancellation_manager,
413 TF_CAPI_EXPORT extern const TFE_OpAttrs* TFE_OpGetAttrs(const TFE_Op* op);
417 TF_CAPI_EXPORT extern void TFE_OpAddAttrs(TFE_Op* op, const TFE_OpAttrs* attrs);
428 TF_CAPI_EXPORT extern void TFE_OpSetAttrValueProto(const TFE_Op* op,
466 void (*execute)(const TFE_Op* op, int* num_outputs,
Dc_api_distributed_test.cc196 TFE_Op* func = TFE_NewOp(ctx, "AddVariablesFunction", status); in TestFunctionWithPackedInput()
387 TFE_Op* func = TFE_NewOp(ctx, "VariableAddFunction", status); in TEST()
511 TFE_Op* func = TFE_NewOp(ctx, "VariableAddFunction", status); in TestDistributedFunctionCancellation()
593 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
Dcustom_device_test.cc49 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> matmul( in TEST()
82 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> reused_op( in TEST()
113 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in TEST()
198 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in TEST()
298 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> matmul( in TEST()
Dcustom_device_testutil.cc90 void LoggingDeviceExecute(const TFE_Op* original_op, int* num_outputs, in LoggingDeviceExecute()
109 TFE_Op* op(TFE_NewOp(context, operation_name, s)); in LoggingDeviceExecute()
Dimmediate_execution_operation.h32 struct TFE_Op;
/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_builder_jni.cc30 TFE_Op* requireOp(JNIEnv* env, jlong handle) { in requireOp()
36 return reinterpret_cast<TFE_Op*>(handle); in requireOp()
73 TFE_Op* op = TFE_NewOp(context, op_or_function_name, status); in Java_org_tensorflow_EagerOperationBuilder_allocate()
80 static_assert(sizeof(jlong) >= sizeof(TFE_Op*), in Java_org_tensorflow_EagerOperationBuilder_allocate()
88 TFE_DeleteOp(reinterpret_cast<TFE_Op*>(op_handle)); in Java_org_tensorflow_EagerOperationBuilder_delete()
93 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_execute()
118 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_setDevice()
130 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_addInput()
142 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_addInputList()
167 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_setAttrString()
[all …]
Deager_operation_jni.cc30 TFE_Op* requireOp(JNIEnv* env, jlong handle) { in requireOp()
36 return reinterpret_cast<TFE_Op*>(handle); in requireOp()
53 TFE_DeleteOp(reinterpret_cast<TFE_Op*>(handle)); in Java_org_tensorflow_EagerOperation_delete()
80 TFE_Op* op = requireOp(env, handle); in Java_org_tensorflow_EagerOperation_outputListLength()
96 TFE_Op* op = requireOp(env, handle); in Java_org_tensorflow_EagerOperation_inputListLength()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_testlib.cc37 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in Create()
62 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in Destroy()
79 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in Read()
98 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in GeneralAssignment()
162 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in ExtractPerDeviceValues()
184 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op( in Multiply()
Dparallel_device_lib_test.cc55 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op( in TEST()
73 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> read_op( in TEST()
111 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op( in TEST()
183 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> call_op( in TEST()
275 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> size_op( in TEST()
Dparallel_device.cc36 void operator()(TFE_Op* to_delete) const { TFE_DeleteOp(to_delete); } in operator ()()
39 using OpPtr = std::unique_ptr<TFE_Op, OpDeleter>;
276 void ParallelDeviceExecute(const TFE_Op* original_op, int* num_outputs, in ParallelDeviceExecute()
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc104 TFE_Op* op, in CheckOutputShapes()
165 TFE_Op* matmul_op; in TEST_F()
205 TFE_Op* reshape_op = TFE_NewOp(tfe_context_, "Reshape", status_); in TEST_F()
216 TFE_Op* fill_op = TFE_NewOp(tfe_context_, "Fill", status_); in TEST_F()
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dsaved_model_api_test.cc123 TFE_Op* compute_fn_op = TF_ConcreteFunctionMakeCallOp( in TEST_P()
262 TFE_Op* serving_default_op = TF_SignatureDefFunctionMakeCallOp( in TEST_P()
319 TFE_Op* read_file_op = in TEST_P()
388 TFE_Op* lookup_op = TF_ConcreteFunctionMakeCallOp( in TEST_P()
420 TFE_Op* lookup_op = TF_ConcreteFunctionMakeCallOp( in TEST_P()
453 TFE_Op* lookup_op = TF_ConcreteFunctionMakeCallOp( in TEST_P()
558 TFE_Op* while_fn_op = TF_ConcreteFunctionMakeCallOp( in TEST_P()
Dsignature_def_function.cc38 TFE_Op* TF_SignatureDefFunctionMakeCallOp(TF_SignatureDefFunction* func, in TF_SignatureDefFunctionMakeCallOp()
Dconcrete_function.cc37 TFE_Op* TF_ConcreteFunctionMakeCallOp(TF_ConcreteFunction* func, in TF_ConcreteFunctionMakeCallOp()
/external/tensorflow/tensorflow/c/experimental/saved_model/public/
Dconcrete_function.h50 TF_CAPI_EXPORT extern TFE_Op* TF_ConcreteFunctionMakeCallOp(
Dsignature_def_function.h42 TF_CAPI_EXPORT extern TFE_Op* TF_SignatureDefFunctionMakeCallOp(
DREADME.md16 TFE_Op* op = TF_ConcreteFunctionMakeCallOp(f, NULL, 0, status);
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc72 void operator()(TFE_Op* op) const { TFE_DeleteOp(op); } in operator ()()
75 std::unique_ptr<TFE_Op, OpDeleter>>
80 std::unique_ptr<TFE_Op, OpDeleter> ReleaseThreadLocalOp(TFE_Context* ctx) { in ReleaseThreadLocalOp()
88 TFE_Op* GetOp(TFE_Context* ctx, const char* op_or_function_name, in GetOp()
102 void ReturnOp(TFE_Context* ctx, TFE_Op* op) { in ReturnOp()
376 TFE_Context* ctx, TFE_Op* op, const char* key, PyObject* py_list, in SetOpAttrList()
483 std::unique_ptr<const TFE_Op*[]> funcs(new const TFE_Op*[num_values]); in SetOpAttrList()
527 TFE_Op* GetFunc(TFE_Context* ctx, const tensorflow::NameAttrList& func, in GetFunc()
529 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc()
539 TFE_Context* ctx, TFE_Op* op, const tensorflow::OpDef::AttrDef& attr, in SetOpAttrListDefault()
[all …]

12