Home
last modified time | relevance | path

Searched refs:TF_Bool (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/c/
Dc_api_macros.h35 #ifndef TF_Bool
36 #define TF_Bool unsigned char macro
Dkernels.h247 TF_OpKernelConstruction* ctx, const char* attr_name, TF_Bool* val,
309 TF_OpKernelConstruction* ctx, const char* attr_name, TF_Bool* vals,
Dtf_tensor.h53 TF_Bool on_host;
Dkernels.cc388 DEFINE_TF_GETATTR(Bool, TF_Bool, bool, "bool", b)
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.h273 TF_Bool (*get_allocator_stats)(const SP_Device* device,
278 TF_Bool (*device_memory_usage)(const SP_Device* device, int64_t* free,
398 TF_Bool (*host_callback)(const SP_Device* device, SP_Stream stream,
436 TF_Bool supports_unified_memory;
Dstream_executor_test.cc57 TF_Bool get_allocator_stats(const SP_Device* const device, in get_allocator_stats()
61 TF_Bool device_memory_usage(const SP_Device* const device, int64_t* const free, in device_memory_usage()
111 TF_Bool host_callback(const SP_Device* const device, SP_Stream stream, in host_callback()
411 SP_AllocatorStats* const stat) -> TF_Bool { in TEST_F()
427 int64_t* const total) -> TF_Bool { in TEST_F()
893 void* const callback_arg) -> TF_Bool { in TEST_F()
913 void* const callback_arg) -> TF_Bool { in TEST_F()
Dstream_executor.cc306 TF_Bool has_stats = in GetAllocatorStats()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_op_base.td268 def TF_Bool : AnyTypeOf<[I<1>, TF_BoolRef], "bool">;
294 def TF_BoolTensor : TensorOf<[TF_Bool]>;
409 TF_Bool.predicate,
Dtf_generated_ops.td253 …Arg<TensorOf<[TF_Bfloat16, TF_Bool, TF_Complex128, TF_Complex64, TF_Float16, TF_Float32, TF_Float6…
264 …Res<TensorOf<[TF_Bfloat16, TF_Bool, TF_Complex128, TF_Complex64, TF_Float16, TF_Float32, TF_Float6…
462 …TensorOf<[TF_Bfloat16, TF_Bool, TF_Complex128, TF_Complex64, TF_Float16, TF_Float32, TF_Float64, T…
497 …TensorOf<[TF_Bfloat16, TF_Bool, TF_Complex128, TF_Complex64, TF_Float16, TF_Float32, TF_Float64, T…
531 …TensorOf<[TF_Bool, TF_Complex128, TF_Complex64, TF_Float32, TF_Float64, TF_Int16, TF_Int32, TF_Int…
1843 TensorOf<[TF_Bool, TF_Float16, TF_Float32, TF_Float64, TF_Int32, TF_Int64]>:$data
1853 TensorOf<[TF_Bool, TF_Float16, TF_Float32, TF_Float64, TF_Int32, TF_Int64]>:$input,
1864 TensorOf<[TF_Bool, TF_Float16, TF_Float32, TF_Float64, TF_Int32, TF_Int64]>:$data
4116 …Arg<TensorOf<[TF_Bfloat16, TF_Bool, TF_Complex128, TF_Complex64, TF_Float16, TF_Float32, TF_Float6…
4125 …Res<TensorOf<[TF_Bfloat16, TF_Bool, TF_Complex128, TF_Complex64, TF_Float16, TF_Float32, TF_Float6…
[all …]