Searched refs:SetOutputAttrs (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 56 void SetOutputAttrs(OpKernelContext::Params* params, in SetOutputAttrs() function
174 test::SetOutputAttrs(params_.get(), &attrs); in RunOpKernel()
|
| D | segment_reduction_ops_test.cc | 81 test::SetOutputAttrs(¶ms, &attrs); in BM_UnsortedSegmentReduction()
144 test::SetOutputAttrs(¶ms, &attrs); in BM_SegmentReduction()
|
| D | ops_testutil.h | 61 void SetOutputAttrs(OpKernelContext::Params* params,
|
| D | restore_op_test.cc | 180 test::SetOutputAttrs(¶ms, &attrs); in TEST_F()
425 test::SetOutputAttrs(¶ms, &attrs); in TEST_F()
|
| D | parse_tensor_test.cc | 63 test::SetOutputAttrs(¶ms, &attrs); in ParseSerializedWithNodeDef()
|
| D | nn_ops_test.cc | 812 test::SetOutputAttrs(¶ms, &attrs); in BM_LRNFloat()
891 test::SetOutputAttrs(¶ms, &attrs); in BM_AvgPool()
994 test::SetOutputAttrs(¶ms, &attrs); in BM_AvgPoolBk()
1085 test::SetOutputAttrs(¶ms, &attrs); in BM_MaxPool()
1267 test::SetOutputAttrs(¶ms, &attrs); in BM_ReluFloat()
1338 test::SetOutputAttrs(¶ms, &attrs); in BM_SoftplusFloat()
|
| D | sparse_to_dense_op_test.cc | 254 test::SetOutputAttrs(¶ms, &attrs); in BM_SparseToDense()
|
| D | restore_v2_op_test.cc | 189 test::SetOutputAttrs(¶ms, &attrs); in RunTest()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 264 void SetOutputAttrs(OpKernelContext::Params* params, in SetOutputAttrs() function in tensorflow::__anon8b4f34150511::EMBenchmarkHelper
316 SetOutputAttrs(params, &allocator_attrs_); in PrepOpKernel()
|