Home
last modified time | relevance | path

Searched refs:params_input (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc928 void RestoreParams(const OpInputList params_input, in RestoreParams() argument
932 CHECK(params_input.size() == num_params) in RestoreParams()
933 << "Number of params mismatch. Expected " << params_input.size() in RestoreParams()
938 CHECK(size == params_input[i].NumElements()) in RestoreParams()
940 << params_input[i].NumElements(); in RestoreParams()
941 auto data_src_ptr = StreamExecutorUtil::AsDeviceMemory<T>(params_input[i]); in RestoreParams()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5318 const auto& params_input = inputs.at(0); in ConvertGather() local
5335 axis[0], params_input.GetTrtDims().nbDims, node_def.name(), in ConvertGather()
5336 params->use_implicit_batch && params_input.is_tensor(), &trt_axis)); in ConvertGather()
5337 if (params->use_implicit_batch && params_input.is_weights() && in ConvertGather()
5342 if (params->use_implicit_batch && params_input.is_tensor() && in ConvertGather()
5352 params_input.GetTrtDims().nbDims + in ConvertGather()
5353 (params->use_implicit_batch && params_input.is_tensor() ? 1 : 0); in ConvertGather()
5367 if (params_input.is_weights()) { in ConvertGather()
5369 params_input.weights(), params_input.GetTrtDims()); in ConvertGather()
5371 params_tensor = params_input.tensor(); in ConvertGather()
[all …]
Dconvert_nodes_test.cc5279 const std::vector<int> params_input = {1, 2, 3, 4, 5, 6}; in TEST_P() local
5448 AddTestTensor("params", p.params_shape, params_input); in TEST_P()
5450 AddTestWeights("params", p.params_shape, params_input, tf_type_); in TEST_P()