Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dhoist_random_uniform.cc149 const FunctionDef* MakeLessStatefulFunction(const FunctionDef& map_function, in MakeLessStatefulFunction() argument
154 *stateless_function = map_function; in MakeLessStatefulFunction()
180 bool CanHoistRandomUniform(const FunctionDef& map_function, in CanHoistRandomUniform() argument
184 if (!map_function.signature().is_stateful()) return false; in CanHoistRandomUniform()
189 for (const auto& node : map_function.node_def()) { in CanHoistRandomUniform()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dfilter_fusion_test.py65 def testFilterFusion(self, map_function, predicates): argument
68 "MemoryCacheImpl"])).map(map_function)
79 r = map_function(x)
/external/tensorflow/tensorflow/compiler/xla/
Dreference_util.h165 const std::function<float(float)>& map_function);
171 const std::function<float(float, float)>& map_function);
437 const std::function<float(float, int64, int64)>& map_function);
443 F&& map_function) { in MapArray4D() argument
446 return map_function(value); in MapArray4D()
456 const Array4D<float>& input, F&& map_function) { in MapWithIndexArray4D() argument
464 map_function(input(plane, depth, height, width), plane, depth, in MapWithIndexArray4D()
478 F&& map_function) { in MapArray4D() argument
481 return map_function(lhs, rhs); in MapArray4D()
491 const Array4D<float>& lhs, const Array4D<float>& rhs, F&& map_function) { in MapWithIndexArray4D() argument
[all …]
Dreference_util.cc702 const std::function<float(float)>& map_function) { in MapArray2D() argument
708 (*result)(i, j) = map_function(matrix(i, j)); in MapArray2D()
716 const std::function<float(float, float)>& map_function) { in MapArray2D() argument
724 (*result)(i, j) = map_function(lhs(i, j), rhs(i, j)); in MapArray2D()
732 const std::function<float(float, int64, int64)>& map_function) { in MapWithIndexArray2D() argument
738 (*result)(i, j) = map_function(matrix(i, j), i, j); in MapWithIndexArray2D()
/external/v8/src/builtins/
Dbuiltins-array-gen.cc2030 TNode<Object> map_function = args.GetOptionalArgumentValue(1); in TF_BUILTIN() local
2035 GotoIf(IsUndefined(map_function), &no_error); in TF_BUILTIN()
2036 GotoIf(TaggedIsSmi(map_function), &error); in TF_BUILTIN()
2037 Branch(IsCallable(CAST(map_function)), &no_error, &error); in TF_BUILTIN()
2040 ThrowTypeError(context, MessageTemplate::kCalledNonCallable, map_function); in TF_BUILTIN()
2112 GotoIf(IsUndefined(map_function), &next); in TF_BUILTIN()
2114 CSA_ASSERT(this, IsCallable(CAST(map_function))); in TF_BUILTIN()
2115 Node* v = CallJS(CodeFactory::Call(isolate()), context, map_function, in TF_BUILTIN()
2190 GotoIf(IsUndefined(map_function), &next); in TF_BUILTIN()
2192 CSA_ASSERT(this, IsCallable(CAST(map_function))); in TF_BUILTIN()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.h1211 std::function<HloComputation*(HloComputation*)> map_function) { in ReplaceCalledComputations() argument
1213 called_computations_[i] = map_function(called_computations_[i]); in ReplaceCalledComputations()