Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/keras/
Dbackend_config_test.py33 keras.backend_config.set_epsilon(epsilon)
34 self.assertEqual(keras.backend_config.epsilon(), epsilon)
35 keras.backend_config.set_epsilon(1e-7)
36 self.assertEqual(keras.backend_config.epsilon(), 1e-7)
40 keras.backend_config.set_floatx(floatx)
41 self.assertEqual(keras.backend_config.floatx(), floatx)
42 keras.backend_config.set_floatx('float32')
43 self.assertEqual(keras.backend_config.floatx(), 'float32')
47 keras.backend_config.set_image_data_format(image_data_format)
48 self.assertEqual(keras.backend_config.image_data_format(),
[all …]
DBUILD48 ":backend_config",
92 name = "backend_config",
93 srcs = ["backend_config.py"],
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_thunk.cc41 const GemmBackendConfig &backend_config) in GemmThunk() argument
47 backend_config_(backend_config) {} in GemmThunk()
159 const GemmBackendConfig &backend_config, in RunGemm() argument
176 const DotDimensionNumbers &dim_nums = backend_config.dot_dimension_numbers(); in RunGemm()
184 int64 batch_size = backend_config.batch_size(); in RunGemm()
257 if (backend_config.algorithm_case() == in RunGemm()
261 return backend_config.selected_algorithm(); in RunGemm()
264 complex128 alpha = {backend_config.alpha_real(), backend_config.alpha_imag()}; in RunGemm()
265 double beta = backend_config.beta(); in RunGemm()
Dgemm_algorithm_picker.cc70 GemmBackendConfig backend_config = in DoUncachedGemmAutotune() local
71 gemm->backend_config<GemmBackendConfig>().ValueOrDie(); in DoUncachedGemmAutotune()
88 if (reinit_cublas_data && backend_config.beta() != 0) { in DoUncachedGemmAutotune()
99 CHECK(RunGemm(gemm, backend_config, lhs_buffer, rhs_buffer, output_buffer, in DoUncachedGemmAutotune()
207 instr->backend_config<GemmBackendConfig>().ValueOrDie(); in DoGemmAutotune()
280 instr->backend_config<GemmBackendConfig>().ValueOrDie(); in RunOnInstruction()
Dgpu_conv_runner.cc257 TF_ASSIGN_OR_RETURN(CudnnConvBackendConfig backend_config, in GetGpuConvParams()
258 conv->backend_config<CudnnConvBackendConfig>()); in GetGpuConvParams()
268 se::dnn::AlgorithmDesc(backend_config.algorithm(), in GetGpuConvParams()
269 backend_config.tensor_ops_enabled()), in GetGpuConvParams()
271 params.conv_result_scale = backend_config.conv_result_scale(); in GetGpuConvParams()
304 if (!se::dnn::ActivationMode_IsValid(backend_config.activation_mode())) { in GetGpuConvParams()
306 backend_config.ShortDebugString()); in GetGpuConvParams()
309 backend_config.activation_mode()); in GetGpuConvParams()
310 fusion.side_input_scale = backend_config.side_input_scale(); in GetGpuConvParams()
Dgemm_thunk.h47 const GemmBackendConfig& backend_config);
72 const HloInstruction* gemm, const GemmBackendConfig& backend_config,
Dgemm_rewriter.cc85 existing_gemm->backend_config<GemmBackendConfig>()); in HandleMultiply()
106 existing_gemm->backend_config<GemmBackendConfig>().ValueOrDie(); in HandleAdd()
Dgpu_conv_algorithm_picker.cc405 TF_ASSIGN_OR_RETURN(auto backend_config, in PickBestAlgorithmNoCacheCuda()
406 instr->backend_config<CudnnConvBackendConfig>()); in PickBestAlgorithmNoCacheCuda()
774 TF_ASSIGN_OR_RETURN(CudnnConvBackendConfig backend_config, in RunOnInstruction()
775 instr->backend_config<CudnnConvBackendConfig>()); in RunOnInstruction()
776 backend_config.set_algorithm(best_algo.conv().algorithm()); in RunOnInstruction()
777 backend_config.set_tensor_ops_enabled(best_algo.conv().tensor_ops_enabled()); in RunOnInstruction()
785 TF_RETURN_IF_ERROR(new_call->set_backend_config(backend_config)); in RunOnInstruction()
Dbackend_configs.proto18 // See HloInstruction::backend_config() for more info.
Dthunk_emitter.cc72 auto config_or = inst->backend_config<GemmBackendConfig>(); in BuildGemmThunk()
242 custom_call->backend_config<CholeskyOptions>()); in HandleCustomCall()
Dcudnn_fused_conv_rewriter_test.cc703 …im_labels=bf01_01io->bf01, custom_call_target="__cudnn$convForward", backend_config="{\"convResult…
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_trip_count_annotator_test.cc61 ->backend_config<WhileLoopBackendConfig>()); in TEST_F()
97 ->backend_config<WhileLoopBackendConfig>()); in TEST_F()
133 ->backend_config<WhileLoopBackendConfig>()); in TEST_F()
169 ->backend_config<WhileLoopBackendConfig>()); in TEST_F()
Dhlo.proto52 // Use backend_config instead for custom_call_opaque.
55 // Use backend_config instead for all_reduce_barrier.
174 string backend_config = 43; field
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgemm_rewrite_test.cc101 …2[2,2]{1,0} custom-call(%x, %y), custom_call_target="__cublas$gemm", backend_config="{\"alpha_real…
125 …2[2,2]{1,0} custom-call(%y, %x), custom_call_target="__cublas$gemm", backend_config="{\"alpha_real…
151 …2[2,2]{1,0} custom-call(%x, %y), custom_call_target="__cublas$gemm", backend_config="{\"alpha_real…
177 …4[2,2]{1,0} custom-call(%x, %y), custom_call_target="__cublas$gemm", backend_config="{\"alpha_real…
200 …2[2,2]{1,0} custom-call(%x, %y), custom_call_target="__cublas$gemm", backend_config="{\"alpha_real…
225 …2[2,2]{1,0} custom-call(%x, %y), custom_call_target="__cublas$gemm", backend_config="{\"alpha_real…
254 …1,0} custom-call(%x, %y, %bias), custom_call_target="__cublas$gemm", backend_config="{\"alpha_real…
284 …2[2,2]{1,0} custom-call(%x, %y), custom_call_target="__cublas$gemm", backend_config="{\"alpha_real…
/external/tensorflow/tensorflow/compiler/xla/service/gpu/data/
Dhlo_algorithm_blacklist.pbtxt2 …im_labels=b01f_01io->b01f, custom_call_target="__cudnn$convForward", backend_config="{conv_result_…
10 …im_labels=b01f_01io->b01f, custom_call_target="__cudnn$convForward", backend_config="{conv_result_…
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Dadagrad.py25 from tensorflow.python.keras import backend_config
93 epsilon = backend_config.epsilon()
98 self.epsilon = epsilon or backend_config.epsilon()
Dadadelta.py24 from tensorflow.python.keras import backend_config
102 self.epsilon = epsilon or backend_config.epsilon()
Dadamax.py23 from tensorflow.python.keras import backend_config
106 self.epsilon = epsilon or backend_config.epsilon()
Dadam.py21 from tensorflow.python.keras import backend_config
147 self.epsilon = epsilon or backend_config.epsilon()
Dnadam.py21 from tensorflow.python.keras import backend_config
102 self.epsilon = epsilon or backend_config.epsilon()
Drmsprop.py23 from tensorflow.python.keras import backend_config
119 self.epsilon = epsilon or backend_config.epsilon()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dhlo_test_base.cc331 string backend_config) { in Run() argument
360 if (!backend_config.empty()) { in Run()
364 instruction->set_raw_backend_config_string(backend_config); in Run()
378 std::vector<ExecutionProfile>* profiles, string backend_config, in RunMultipleTimes() argument
413 if (!backend_config.empty()) { in RunMultipleTimes()
417 instruction->set_raw_backend_config_string(backend_config); in RunMultipleTimes()
Dhlo_test_base.h217 string backend_config = "") TF_MUST_USE_RESULT;
223 std::vector<ExecutionProfile>* profiles, string backend_config = "",
/external/tensorflow/tensorflow/python/keras/premade/
DBUILD26 "//tensorflow/python/keras:backend_config",
/external/tensorflow/tensorflow/python/keras/api/
DBUILD30 "tensorflow.python.keras.backend_config",

12