Home
last modified time | relevance | path

Searched defs:output_tensor (Results 1 – 25 of 109) sorted by relevance

12345

/external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/
Dmultiple_additive_trees_test.cc58 auto output_tensor = AsTensor<float>({9.0f, 23.0f}, {2, 1}); in TEST_F() local
94 auto output_tensor = AsTensor<float>({0.0f, 0.0f}, {2, 1}); in TEST_F() local
191 auto output_tensor = AsTensor<float>({0.0f, 0.0f, 0.0f, 0.0f}, {2, 2}); in TEST_F() local
277 auto output_tensor = in TEST_F() local
/external/tensorflow/tensorflow/core/kernels/
Dbase64_ops.cc35 Tensor* output_tensor = nullptr; in Compute() local
60 Tensor* output_tensor = nullptr; in Compute() local
Dregex_full_match_op.cc47 Tensor* output_tensor = nullptr; in Compute() local
76 Tensor* output_tensor = nullptr; in Compute() local
Dstring_to_hash_bucket_op.h41 Tensor* output_tensor = nullptr; in Compute() local
83 Tensor* output_tensor = nullptr; in Compute() local
Ddecode_raw_op.cc55 Tensor* output_tensor = nullptr; in Compute() local
67 Tensor* output_tensor = nullptr; in Compute() local
Ddebug_ops.h165 Tensor* output_tensor; in ApplyGrpcGating() local
236 Tensor* output_tensor; in Compute() local
279 Tensor* output_tensor; in Compute() local
Dstring_strip_op.cc36 Tensor* output_tensor; in Compute() local
Dstring_to_number_op.cc45 Tensor* output_tensor = nullptr; in Compute() local
Dunicode_script_op.cc31 Tensor* output_tensor = nullptr; in Compute() local
Dstring_to_hash_bucket_op.cc38 Tensor* output_tensor = nullptr; in Compute() local
Dstring_join_op.cc60 Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc305 Tensor* output_tensor; in PopulateOutputs() local
356 Tensor* output_tensor; in PopulateOutputs() local
370 Tensor output_tensor = XlaTensorBuffer::MakeTensor( in PopulateOutputs() local
419 Tensor output_tensor; in PopulateOutputs() local
434 Tensor output_tensor = XlaTensorBuffer::MakeTensor( in PopulateOutputs() local
/external/tensorflow/tensorflow/core/user_ops/
Dfact.cc33 tensorflow::Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/python/kernel_tests/
Dackermann_op.cc35 Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/lite/experimental/c/
Dc_api_test.cc73 const TFL_Tensor* output_tensor = in TEST() local
132 const TFL_Tensor* output_tensor = in TEST() local
/external/tensorflow/tensorflow/examples/adding_an_op/
Dcuda_op_kernel.cc43 Tensor* output_tensor = nullptr; in Compute() local
Dzero_out_op_kernel_1.cc46 Tensor* output_tensor = nullptr; in Compute() local
Dzero_out_op_kernel_3.cc53 Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/tools/ci_build/builds/user_ops/
Dcuda_op_kernel.cc43 Tensor* output_tensor = NULL; in Compute() local
Dzero_out_op_kernel_1.cc46 Tensor* output_tensor = NULL; in Compute() local
/external/tensorflow/tensorflow/c/
Dc_test_util.h149 TF_Tensor* output_tensor(int i) { return output_values_[i]; } in output_tensor() function
/external/tensorflow/tensorflow/contrib/input_pipeline/kernels/
Dinput_pipeline_kernels.cc36 Tensor* output_tensor = nullptr; in Compute() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc1156 const nvinfer1::ITensor** output_tensor) { in TransposeTensor()
1749 const nvinfer1::ITensor* output_tensor = layer->getOutput(0); in BinaryTensorOpWeight() local
1922 const nvinfer1::ITensor* output_tensor = conv_layer->getOutput(0); in ConvertConv2DHelper() local
1997 nvinfer1::ITensor* output_tensor = layer->getOutput(0); in BinaryTensorOpTensor() local
2037 nvinfer1::ITensor* output_tensor = layer->getOutput(i); in ConvertPlugin() local
2070 const nvinfer1::ITensor* output_tensor = nullptr; in ConvertTranspose() local
2169 const nvinfer1::ITensor* output_tensor = nullptr; in ConvertReshape() local
2222 const nvinfer1::ITensor* output_tensor = nullptr; in ConvertExpandDims() local
2283 const nvinfer1::ITensor* output_tensor = nullptr; in ConvertSqueeze() local
2437 const nvinfer1::ITensor* output_tensor = nullptr; in ConvertStridedSliceHelper() local
[all …]
/external/eigen/unsupported/test/
Dcxx11_tensor_forced_eval.cpp65 …Tensor<float, 2> output_tensor= (input_tensor - input_tensor.maximum(depth_dim).eval().reshape(dim… in test_const() local
/external/tensorflow/tensorflow/lite/tools/optimize/
Dsubgraph_quantizer.cc275 auto output_tensor = subgraph_->tensors[op->outputs[0]].get(); in PropagateMinMaxForAvgAndMaxPool() local
319 auto output_tensor = subgraph_->tensors[op->outputs[0]].get(); in AsymmetricQuantizeSoftmax() local
345 auto output_tensor = subgraph_->tensors[op->outputs[output_idx]].get(); in AsymmetricQuantizeInputsAndOutputs() local

12345