Home
last modified time | relevance | path

Searched refs:SetOutputAttrs (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h52 inline void SetOutputAttrs(OpKernelContext::Params* params, in SetOutputAttrs() function
175 test::SetOutputAttrs(params_.get(), &attrs); in RunOpKernel()
Drestore_op_test.cc177 test::SetOutputAttrs(&params, &attrs); in TEST_F()
422 test::SetOutputAttrs(&params, &attrs); in TEST_F()
Dsegment_reduction_ops_test.cc77 test::SetOutputAttrs(&params, &attrs); in BM_SegmentReduction()
Dparse_tensor_test.cc60 test::SetOutputAttrs(&params, &attrs); in ParseSerializedWithNodeDef()
Dsparse_to_dense_op_test.cc256 test::SetOutputAttrs(&params, &attrs); in BM_SparseToDense()
Dnn_ops_test.cc771 test::SetOutputAttrs(&params, &attrs); in BM_LRNFloat()
851 test::SetOutputAttrs(&params, &attrs); in BM_AvgPool()
956 test::SetOutputAttrs(&params, &attrs); in BM_AvgPoolBk()
1046 test::SetOutputAttrs(&params, &attrs); in BM_MaxPool()
1222 test::SetOutputAttrs(&params, &attrs); in BM_ReluFloat()
Drestore_v2_op_test.cc186 test::SetOutputAttrs(&params, &attrs); in RunTest()