| /external/tensorflow/tensorflow/python/keras/ |
| D | backend_config_test.py | 33 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 …]
|
| D | backend.py | 47 from tensorflow.python.keras import backend_config
120 epsilon = backend_config.epsilon
121 floatx = backend_config.floatx
122 image_data_format = backend_config.image_data_format
123 set_epsilon = backend_config.set_epsilon
124 set_floatx = backend_config.set_floatx
125 set_image_data_format = backend_config.set_image_data_format
|
| D | BUILD | 86 ":backend_config",
127 name = "backend_config",
128 srcs = ["backend_config.py"],
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_trip_count_annotator_test.cc | 61 ->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()
|
| D | hlo_parser_test.cc | 87 …"\"it\'s not a problem\n" source_file="path/to/test.cc" source_line=68}, backend_config="foo\" bar" in CreateTestCases()
122 ROOT %constant = f32[] constant(42), backend_config="this is a configuration" in CreateTestCases()
1698 %constant = s32[] constant(42), backend_config="foo bar" in CreateTestCases()
1905 …2,1,0} %filter), feature_group_count=1, sharding={maximal device=1}, backend_config="foo", dim_lab… in CreateTestCases()
|
| D | hlo.proto | 170 string backend_config = 43; field
|
| D | hlo_parser.cc | 680 optional<string> backend_config; in ParseInstructionRhs() local
682 &backend_config}; in ParseInstructionRhs()
1749 if (backend_config) { in ParseInstructionRhs()
1750 instruction->set_raw_backend_config_string(std::move(*backend_config)); in ParseInstructionRhs()
|
| D | hlo_instruction.h | 1304 StatusOr<ConfigProto> backend_config() const { in backend_config() function
|
| D | hlo_instruction.cc | 620 instruction->backend_config_ = proto.backend_config(); in CreateFromProto()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_runner.cc | 326 TF_ASSIGN_OR_RETURN(CudnnConvBackendConfig backend_config, in GetCudnnConvParams()
327 conv->backend_config<CudnnConvBackendConfig>()); in GetCudnnConvParams()
338 backend_config.algorithm(), backend_config.tensor_ops_enabled())); in GetCudnnConvParams()
339 params.conv_result_scale = backend_config.conv_result_scale(); in GetCudnnConvParams()
373 if (!se::dnn::ActivationMode_IsValid(backend_config.activation_mode())) { in GetCudnnConvParams()
375 backend_config.ShortDebugString()); in GetCudnnConvParams()
378 backend_config.activation_mode()); in GetCudnnConvParams()
379 fusion.side_input_scale = backend_config.side_input_scale(); in GetCudnnConvParams()
|
| D | cudnn_conv_algorithm_picker.cc | 225 TF_ASSIGN_OR_RETURN(auto backend_config, in PickBestAlgorithm()
226 instr->backend_config<CudnnConvBackendConfig>()); in PickBestAlgorithm()
383 TF_ASSIGN_OR_RETURN(CudnnConvBackendConfig backend_config, in RunOnInstruction()
384 instr->backend_config<CudnnConvBackendConfig>()); in RunOnInstruction()
385 backend_config.set_algorithm(best_algo.conv().algorithm()); in RunOnInstruction()
386 backend_config.set_tensor_ops_enabled(best_algo.conv().tensor_ops_enabled()); in RunOnInstruction()
394 TF_RETURN_IF_ERROR(new_call->set_backend_config(backend_config)); in RunOnInstruction()
|
| D | backend_configs.proto | 16 // See HloInstruction::backend_config() for more info.
|
| D | cudnn_fused_conv_rewriter.cc | 231 conv->backend_config<CudnnConvBackendConfig>()); in TryRewriteToCudnnForwardRelu()
|
| D | ir_emitter_unnested.cc | 486 custom_call->backend_config<CholeskyOptions>()); in HandleCustomCall()
983 auto config = xla_while->backend_config<WhileLoopBackendConfig>(); in HandleWhile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.cc | 345 string backend_config) { in Run() argument
375 if (!backend_config.empty()) { in Run()
379 instruction->set_raw_backend_config_string(backend_config); in Run()
394 std::vector<ExecutionProfile>* profiles, string backend_config) { in RunMultipleTimes() argument
429 if (!backend_config.empty()) { in RunMultipleTimes()
433 instruction->set_raw_backend_config_string(backend_config); in RunMultipleTimes()
|
| D | hlo_test_base.h | 242 string backend_config = "") TF_MUST_USE_RESULT;
246 string backend_config = "") TF_MUST_USE_RESULT;
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | adadelta.py | 23 from tensorflow.python.keras import backend_config
96 epsilon = backend_config.epsilon()
|
| D | adagrad.py | 24 from tensorflow.python.keras import backend_config
94 epsilon = backend_config.epsilon()
|
| D | adamax.py | 22 from tensorflow.python.keras import backend_config
100 epsilon = backend_config.epsilon()
|
| D | adam.py | 21 from tensorflow.python.keras import backend_config
134 epsilon = backend_config.epsilon()
|
| D | nadam.py | 21 from tensorflow.python.keras import backend_config
97 epsilon = backend_config.epsilon()
|
| D | rmsprop.py | 23 from tensorflow.python.keras import backend_config
107 epsilon = backend_config.epsilon()
|
| D | BUILD | 39 "//tensorflow/python/keras:backend_config",
|