Home
last modified time | relevance | path

Searched defs:weights_offset (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv.cc121 std::string weights_offset = "args.kernel_size_x * args.kernel_size_y"; in GenerateDepthwiseConvolutionCode() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h863 int32_t weights_offset; member
903 int32_t weights_offset; member
937 int32_t weights_offset; member
/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_rnn_test.cc1434 int weights_offset = 0; in run_closedbox_test_with_input_split() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4159 const int weights_offset = i * input_dimensions.NodesAcrossFeatureMaps() * in DoMatMul() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3912 const int weights_offset = i * input_dimensions.NodesAcrossFeatureMaps() * in DoMatMul() local