Home
last modified time | relevance | path

Searched refs:add_func (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsoftmax_op.cc47 const xla::Computation& add_func = *ctx->GetOrCreateAdd(type); in Compile() local
60 add_func, {kClassDim})); in Compile()
65 auto sum_exp = b->Reduce(exp_shifted, XlaHelpers::Zero(b, type), add_func, in Compile()
85 const xla::Computation& add_func = *ctx->GetOrCreateAdd(type); in CrossEntropyWithLogits() local
104 add_func, {kClassDim}); in CrossEntropyWithLogits()
115 XlaHelpers::Zero(b, type), add_func, {kClassDim}); in CrossEntropyWithLogits()
/external/autotest/server/
Dsite_utils.py694 def add_func(func): function
702 return add_func
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc1234 auto add_func = module().AddEmbeddedComputation(add_computation.Build()); in TEST_P() local
1239 /*dimensions_to_reduce=*/{1}, add_func)); in TEST_P()
1327 auto add_func = module().AddEmbeddedComputation(add_computation.Build()); in TEST_P() local
1348 shape, arg_instruction, init_value, window, add_func)); in TEST_P()
1380 auto add_func = module().AddEmbeddedComputation(add_computation.Build()); in TEST_P() local
1409 shape, arg_instruction, init_value, window, add_func)); in TEST_P()
/external/tensorflow/tensorflow/core/framework/
Dattr_value_util_test.cc53 NameAttrList* entry = ret.mutable_list()->add_func(); in Fs()
Dattr_value_util.cc409 *out->mutable_list()->add_func() = v; in SetAttrValue()
/external/python/cpython3/Modules/
D_pickle.c5964 PyObject *add_func; in load_additems() local
5967 add_func = _PyObject_GetAttrId(set, &PyId_add); in load_additems()
5968 if (add_func == NULL) in load_additems()
5975 result = _Pickle_FastCall(add_func, item); in load_additems()