Searched defs:output_map (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | map_kernels.h | 44 TensorMap** output_map) { in ForwardInputOrCreateNewMap() 138 TensorMap* output_map = nullptr; in Compute() local 160 TensorMap* output_map = nullptr; in Compute() local
|
D | remote_fused_graph_execute_utils.cc | 1179 std::unordered_map<int, string> output_map; in FuseRemoteGraphByPlacedArguments() local
|
/external/mesa3d/src/gallium/drivers/svga/ |
D | svga_link.h | 18 ubyte output_map[PIPE_MAX_SHADER_OUTPUTS]; member
|
D | svga_tgsi_emit.h | 108 SVGA3dShaderDestToken output_map[PIPE_MAX_ATTRIBS]; member
|
/external/tensorflow/tensorflow/tools/graph_transforms/ |
D | summarize_graph_main.cc | 177 std::map<string, std::vector<const NodeDef*>> output_map; in SummarizeGraph() local
|
/external/tensorflow/tensorflow/core/kernels/sparse/ |
D | mat_mul_op.cc | 323 MatrixMap output_map( in SparseDenseMatMulWithoutTransposedLHS() local 396 MatrixMap output_map( in SparseDenseMatMulWithTransposedLHS() local
|
/external/mesa3d/src/freedreno/vulkan/ |
D | tu_shader.c | 699 uint8_t output_map[VARYING_SLOT_TESS_MAX]; in tu_gather_xfb_info() local
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | matrix.cc | 383 absl::flat_hash_set<int64> output_map; in Einsum() local
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | optimized_ops.h | 2147 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 …]
|