Home
last modified time | relevance | path

Searched defs:output_map (Results 1 – 15 of 15) 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
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dgelu.h61 auto output_map = VectorType::Map(output_data, output_shape.FlatSize()); in Gelu() local
/external/mesa3d/src/gallium/drivers/svga/
Dsvga_link.h25 uint8_t output_map[PIPE_MAX_SHADER_OUTPUTS]; member
Dsvga_tgsi_emit.h91 SVGA3dShaderDestToken output_map[PIPE_MAX_ATTRIBS]; member
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
Dtf_jitrt_symbolic_shape_optimization.cc263 AffineMap output_map = AffineMap::getMultiDimIdentityMap(rank, ctx); in matchAndRewrite() local
/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/lite/experimental/acceleration/mini_benchmark/
Dvalidator.cc65 std::map<std::string, std::vector<char>>& output_map) { in AddTensorDataToMap()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dmat_mul_op.cc332 MatrixMap output_map( in SparseDenseMatMulWithoutTransposedLHS() local
405 MatrixMap output_map( in SparseDenseMatMulWithTransposedLHS() local
/external/harfbuzz_ng/src/
Dhb-ot-var-hvar-table.hh182 hb_vector_t<uint32_t> output_map; member
Dhb-ot-layout-common.hh3546 const hb_array_t<const uint32_t> output_map = plan.get_output_map (); in serialize() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1820 auto output_map = MapAsVector(output_data, output_shape); in Add() local
1981 auto output_map = MapAsVector(output_data, output_shape); in MulNoActivation() local
2407 auto output_map = MapAsVector(output_data, output_shape); in SubWithActivation() local
2436 auto output_map = MapAsVector(output_data, output_shape); in Sub() local
3856 auto output_map = MapAsVector(output_data, output_shape); in Logistic() local
3970 auto output_map = MapAsVector(output_data, output_shape); in Tanh() local
4155 auto output_map = MapAsVector(output_data, output_shape); in Cast() local
4163 auto output_map = MapAsVector(output_data, output_shape); in Floor() local
4171 auto output_map = MapAsVector(output_data, output_shape); in Ceil() local
4784 auto output_map = MapAsVector(output_data, output_shape); in Minimum() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmatrix.cc416 absl::flat_hash_set<int64_t> output_map; in Einsum() local
/external/crosvm/devices/src/virtio/video/decoder/backend/
Dvaapi.rs544 let mut output_map = BufferMapping::new(&handle.0, 0, buffer_size)?; in output_picture() localVariable
/external/mesa3d/src/freedreno/vulkan/
Dtu_shader.cc1111 uint8_t output_map[VARYING_SLOT_TESS_MAX]; in tu_gather_xfb_info() local
/external/harfbuzz_ng/src/OT/Color/COLR/
DCOLR.hh2073 hb_vector_t<uint32_t> output_map; member