Home
last modified time | relevance | path

Searched defs:output_map (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmap_kernels.h44 TensorMap** output_map) { in ForwardInputOrCreateNewMap()
138 TensorMap* output_map = nullptr; in Compute() local
160 TensorMap* output_map = nullptr; in Compute() local
Dremote_fused_graph_execute_utils.cc1179 std::unordered_map<int, string> output_map; in FuseRemoteGraphByPlacedArguments() local
/external/mesa3d/src/gallium/drivers/svga/
Dsvga_link.h18 ubyte output_map[PIPE_MAX_SHADER_OUTPUTS]; member
Dsvga_tgsi_emit.h108 SVGA3dShaderDestToken output_map[PIPE_MAX_ATTRIBS]; member
/external/tensorflow/tensorflow/tools/graph_transforms/
Dsummarize_graph_main.cc177 std::map<string, std::vector<const NodeDef*>> output_map; in SummarizeGraph() local
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc323 MatrixMap output_map( in SparseDenseMatMulWithoutTransposedLHS() local
396 MatrixMap output_map( in SparseDenseMatMulWithTransposedLHS() local
/external/mesa3d/src/freedreno/vulkan/
Dtu_shader.c699 uint8_t output_map[VARYING_SLOT_TESS_MAX]; in tu_gather_xfb_info() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc383 absl::flat_hash_set<int64> output_map; in Einsum() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2147 auto output_map = MapAsVector(output_data, output_shape); in Add() local
2308 auto output_map = MapAsVector(output_data, output_shape); in MulNoActivation() local
2730 auto output_map = MapAsVector(output_data, output_shape); in SubWithActivation() local
2759 auto output_map = MapAsVector(output_data, output_shape); in Sub() local
4175 auto output_map = MapAsVector(output_data, output_shape); in Logistic() local
4289 auto output_map = MapAsVector(output_data, output_shape); in Tanh() local
4474 auto output_map = MapAsVector(output_data, output_shape); in Cast() local
4482 auto output_map = MapAsVector(output_data, output_shape); in Floor() local
4490 auto output_map = MapAsVector(output_data, output_shape); in Ceil() local
5546 auto output_map = MapAsVector(output_data, output_shape); in Minimum() local
[all …]