Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms.cc91 auto weights_vector = weights.flat<float>(); in FoldBatchNorms() local
94 for (int64_t row = 0; row < weights_vector.dimension(0); ++row) { in FoldBatchNorms()
96 weights_vector(row) * in FoldBatchNorms()
Dfold_old_batch_norms.cc128 auto weights_vector = weights.flat<float>(); in FuseScaleOffsetToConvWeights() local
131 for (int64_t row = 0; row < weights_vector.dimension(0); ++row) { in FuseScaleOffsetToConvWeights()
133 weights_vector(row) * scale_values[row % weights_cols]; in FuseScaleOffsetToConvWeights()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayerQuantized.cpp133 std::vector<const ICLTensor *> weights_vector; in configure() local
134 weights_vector.emplace_back(&_recurrent_weights); in configure()
135 weights_vector.emplace_back(&_input_weights); in configure()
138 _concat_weights.configure(compile_context, weights_vector, &_weights, Window::DimX); in configure()
349 std::vector<const ITensorInfo *> weights_vector; in validate() local
350 weights_vector.emplace_back(&recurrent_weights); in validate()
351 weights_vector.emplace_back(&input_weights); in validate()
353 … ARM_COMPUTE_RETURN_ON_ERROR(CLConcatenateLayer::validate(weights_vector, &weights, Window::DimX)); in validate()
DCLLSTMLayer.cpp125 std::vector<const ICLTensor *> weights_vector; in configure() local
127 weights_vector.emplace_back(input_to_forget_weights); in configure()
128 weights_vector.emplace_back(recurrent_to_forget_weights); in configure()
129 …concat_shape = arm_compute::misc::shape_calculator::calculate_concatenate_shape(weights_vector, 0); in configure()
132 …_concat_weights_forget_gate.configure(compile_context, weights_vector, &_forget_gate_out6, Window:… in configure()
/external/ComputeLibrary/src/runtime/NEON/functions/
DNELSTMLayerQuantized.cpp112 std::vector<const ITensor *> weights_vector{ &_recurrent_weights, &_input_weights }; in configure() local
114 _concat_weights.configure(weights_vector, &_weights, Window::DimX); in configure()
316 std::vector<const ITensorInfo *> weights_vector; in validate() local
317 weights_vector.emplace_back(&recurrent_weights); in validate()
318 weights_vector.emplace_back(&input_weights); in validate()
320 … ARM_COMPUTE_RETURN_ON_ERROR(NEConcatenateLayer::validate(weights_vector, &weights, Window::DimX)); in validate()
DNELSTMLayer.cpp111 std::vector<const ITensor *> weights_vector; in configure() local
113 weights_vector.emplace_back(input_to_forget_weights); in configure()
114 weights_vector.emplace_back(recurrent_to_forget_weights); in configure()
116 _concat_weights_forget_gate.configure(weights_vector, &_forget_gate_out6, Window::DimX); in configure()