| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.cc | 79 return LookupOrCreate(type, &max_func_, [type] { in GetOrCreateMax()
95 return LookupOrCreate(type, &min_func_, [type] { in GetOrCreateMin()
111 return LookupOrCreate(type, &add_func_, [type] { in GetOrCreateAdd()
127 return LookupOrCreate(type, &mul_func_, [type] { in GetOrCreateMul()
142 const xla::XlaComputation* XlaContext::LookupOrCreate( in LookupOrCreate() function in tensorflow::XlaContext
|
| D | xla_context.h | 119 const xla::XlaComputation* LookupOrCreate(
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_mgr_test.cc | 63 string LookupOrCreate(ResourceMgr* rm, const string& container, in LookupOrCreate() function
66 TF_CHECK_OK(rm->LookupOrCreate<T>(container, name, &r, [&label](T** ret) { in LookupOrCreate()
130 EXPECT_EQ("R/cat", LookupOrCreate<Resource>(&rm, "foo", "bar", "cat")); in TEST()
131 EXPECT_EQ("R/cat", LookupOrCreate<Resource>(&rm, "foo", "bar", "dog")); in TEST()
134 EXPECT_EQ("O/tiger", LookupOrCreate<Other>(&rm, "foo", "bar", "tiger")); in TEST()
135 EXPECT_EQ("O/tiger", LookupOrCreate<Other>(&rm, "foo", "bar", "lion")); in TEST()
148 TF_CHECK_OK(rm.LookupOrCreate<Resource>( in TEST()
|
| D | resource_op_kernel.h | 82 mgr->LookupOrCreate<T>(cinfo_.container(), cinfo_.name(), &resource, in Compute()
|
| D | resource_mgr.h | 159 Status LookupOrCreate(const string& container, const string& name,
493 Status ResourceMgr::LookupOrCreate(const string& container, const string& name,
606 return ctx->resource_manager()->LookupOrCreate(p.container(), p.name(), value,
|
| /external/tensorflow/tensorflow/core/kernels/lookup_tables/ |
| D | op_kernel_templates.h | 92 cinfo_.resource_manager()->template LookupOrCreate<ResourceType, true>( in Compute()
139 return cinfo_.resource_manager()->template LookupOrCreate<T, true>( in RegisterAlias()
227 cinfo_.resource_manager()->template LookupOrCreate<ResourceType, true>( in Compute()
276 return cinfo_.resource_manager()->template LookupOrCreate<T, true>( in RegisterAlias()
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/test_kernels/ |
| D | bigtable_test_client_op.cc | 46 mgr->LookupOrCreate<BigtableClientResource>( in Compute()
|
| /external/tensorflow/tensorflow/contrib/bigtable/kernels/ |
| D | bigtable_kernels.cc | 74 mgr->LookupOrCreate<BigtableClientResource>( in Compute()
150 ctx, mgr->LookupOrCreate<BigtableTableResource>( in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conditional_accumulator_base_op.h | 101 ->template LookupOrCreate<ConditionalAccumulatorBase>( in SetAccumulatorHandle()
|
| D | batch_kernels.cc | 733 c->resource_manager()->LookupOrCreate( in ComputeAsync()
813 c->resource_manager()->LookupOrCreate( in ComputeAsync()
1073 c->resource_manager()->LookupOrCreate( in ComputeAsync()
1259 c->resource_manager()->LookupOrCreate( in ComputeAsync()
|
| D | variable_ops.cc | 68 OP_REQUIRES_OK(ctx, cinfo_.resource_manager()->LookupOrCreate<LegacyVar>( in Compute()
|
| D | lookup_table_op.h | 77 ->template LookupOrCreate<lookup::LookupInterface>( in Compute()
|
| D | stage_op.cc | 194 TF_RETURN_IF_ERROR(rm->LookupOrCreate<Buffer>(cinfo.container(), cinfo.name(), in GetBuffer()
|
| D | conv_ops_fused_image_transform.cc | 348 OP_REQUIRES_OK(context, context->resource_manager()->LookupOrCreate( in operator ()()
386 OP_REQUIRES_OK(context, context->resource_manager()->LookupOrCreate( in operator ()()
|
| D | conv_ops_using_gemm.cc | 314 OP_REQUIRES_OK(context, context->resource_manager()->LookupOrCreate( in operator ()()
|
| D | meta_support.cc | 70 Status s = context->resource_manager()->LookupOrCreate( in GetScratch()
|
| D | quantized_conv_ops.cc | 281 OP_REQUIRES_OK(context, context->resource_manager()->LookupOrCreate( in operator ()()
|
| D | sparse_tensors_map_ops.cc | 148 cinfo_.resource_manager()->LookupOrCreate<SparseTensorsMap>( in GetMap()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 172 TF_RETURN_IF_ERROR(rm->LookupOrCreate<XlaCompilationCache>( in Compile()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_compile_ops.cc | 164 rm->LookupOrCreate<XRTCompilationCache>( in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 297 res_mgr->LookupOrCreate( in ExecuteCalibration()
528 auto status = ctx->resource_manager()->LookupOrCreate( in GetEngine()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | threadpool_dataset_op.cc | 96 OP_REQUIRES_OK(ctx, mgr->LookupOrCreate<ThreadPoolResource>( in Compute()
|
| D | indexed_dataset_op.cc | 281 mgr->LookupOrCreate<MaterializedDatasetResource>( in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 442 mgr->LookupOrCreate<IteratorResource>( in Compute()
847 ctx->resource_manager()->LookupOrCreate<IteratorResource>( in TryInit()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 269 TF_RETURN_IF_ERROR(rm->LookupOrCreate<XlaCompilationCache>( in CompileToLocalExecutable()
|