Searched refs:map_function (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | hoist_random_uniform.cc | 160 const FunctionDef* MakeLessStatefulFunction(const FunctionDef& map_function, in MakeLessStatefulFunction() argument
165 *stateless_function = map_function; in MakeLessStatefulFunction()
190 bool CanHoistRandomUniform(const FunctionDef& map_function, in CanHoistRandomUniform() argument
194 if (!map_function.signature().is_stateful()) return false; in CanHoistRandomUniform()
199 for (const auto& node : map_function.node_def()) { in CanHoistRandomUniform()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | reference_util.h | 165 const std::function<float(float)>& map_function);
171 const std::function<float(float, float)>& map_function);
429 const std::function<float(float, int64, int64)>& map_function);
435 F&& map_function) { in MapArray4D() argument
438 return map_function(value); in MapArray4D()
448 const Array4D<float>& input, F&& map_function) { in MapWithIndexArray4D() argument
456 map_function(input(plane, depth, height, width), plane, depth, in MapWithIndexArray4D()
470 F&& map_function) { in MapArray4D() argument
473 return map_function(lhs, rhs); in MapArray4D()
483 const Array4D<float>& lhs, const Array4D<float>& rhs, F&& map_function) { in MapWithIndexArray4D() argument
[all …]
|
| D | reference_util.cc | 637 const std::function<float(float)>& map_function) { in MapArray2D() argument
643 (*result)(i, j) = map_function(matrix(i, j)); in MapArray2D()
651 const std::function<float(float, float)>& map_function) { in MapArray2D() argument
659 (*result)(i, j) = map_function(lhs(i, j), rhs(i, j)); in MapArray2D()
667 const std::function<float(float, int64, int64)>& map_function) { in MapWithIndexArray2D() argument
673 (*result)(i, j) = map_function(matrix(i, j), i, j); in MapWithIndexArray2D()
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/ |
| D | grappler_test.py | 68 def map_function(x, y): function
76 dataset = dataset.map(map_function)
|
| D | map_vectorization_test.py | 575 def map_function(x): function
581 map_function, num_parallel_calls=10, deterministic=local_determinism)
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | map_test.py | 1353 def map_function(x): function
1362 map_function,
1377 def map_function(x, y): function
1382 dataset = apply_map(dataset, map_function)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction.h | 1459 std::function<HloComputation*(HloComputation*)> map_function) { in ReplaceCalledComputations() argument
1461 called_computations_[i] = map_function(called_computations_[i]); in ReplaceCalledComputations()
|