| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.h | 74 CHECK_NOTNULL(strides); in GetStridesInMklOrder()
91 CHECK_NOTNULL(dilations); in GetDilationsInMklOrder()
117 CHECK_NOTNULL(input_dims); in GetInputSizeInMklOrder()
190 CHECK_NOTNULL(filter_dims); in GetFilterSizeInMklOrder()
284 CHECK_NOTNULL(filter_dims); in GetFilterSizeInMklOrder()
318 CHECK_NOTNULL(output_dims_tf_order);
319 CHECK_NOTNULL(output_dims_mkl_order);
320 CHECK_NOTNULL(pad_l);
321 CHECK_NOTNULL(pad_r);
481 CHECK_NOTNULL(output_dims_tf_order); in GetOutputAndPadSizeInMklOrder()
[all …]
|
| D | remote_fused_graph_execute_utils.cc | 37 CHECK_NOTNULL(node); in FindNodeByName()
78 CHECK_NOTNULL(tid); in IsSameNodeName()
99 CHECK_NOTNULL(str); in AppendDeliminator()
107 CHECK_NOTNULL(out); in ConvertMapToVector()
429 CHECK_NOTNULL(node_def); in GetOutputTensorShapeType()
611 CHECK_NOTNULL(execute_info); in BuildRemoteFusedGraphExecuteInfo()
612 CHECK_NOTNULL(input_types); in BuildRemoteFusedGraphExecuteInfo()
613 CHECK_NOTNULL(output_types); in BuildRemoteFusedGraphExecuteInfo()
680 CHECK_NOTNULL(graph); in BuildRemoteFusedGraphExecuteOpNode()
681 CHECK_NOTNULL(created_node); in BuildRemoteFusedGraphExecuteOpNode()
[all …]
|
| D | mkl_lrn_op.cc | 154 CHECK_NOTNULL(output_tensor); in Compute()
199 CHECK_NOTNULL(output_tensor); in AllocateOutputTensor()
242 CHECK_NOTNULL(output_dnn_data); in MklDefaultToEigen()
251 CHECK_NOTNULL(workspace_tensor); in MklDefaultToEigen()
271 CHECK_NOTNULL(dnn_data_wksp); in AllocateWorkspaceTensor()
284 CHECK_NOTNULL(workspace_tensor); in AllocateWorkspaceTensor()
417 CHECK_NOTNULL(output_tensor); in Compute()
449 CHECK_NOTNULL(output_tensor); in AllocateOutputTensor()
472 CHECK_NOTNULL(input_grad_dnn_data); in ConfigureInputGradient()
514 CHECK_NOTNULL(workspace_dnn_data); in ConfigureWorkspace()
|
| D | mkl_fused_batch_norm_op.cc | 738 CHECK_NOTNULL(dst_tensor); in HandleEmptyInput()
745 CHECK_NOTNULL(*dst_tensor); in HandleEmptyInput()
763 CHECK_NOTNULL(batch_mean_tensor); in AllocateTFOutputs()
764 CHECK_NOTNULL(batch_variance_tensor); in AllocateTFOutputs()
765 CHECK_NOTNULL(saved_mean_tensor); in AllocateTFOutputs()
766 CHECK_NOTNULL(saved_variance_tensor); in AllocateTFOutputs()
778 CHECK_NOTNULL(*batch_mean_tensor); in AllocateTFOutputs()
790 CHECK_NOTNULL(*batch_variance_tensor); in AllocateTFOutputs()
801 CHECK_NOTNULL(*saved_mean_tensor); in AllocateTFOutputs()
811 CHECK_NOTNULL(*saved_variance_tensor); in AllocateTFOutputs()
[all …]
|
| D | mkl_pooling_ops_common.h | 573 CHECK_NOTNULL(output_tensor); in AllocateEmptyOutputTensor()
609 CHECK_NOTNULL(pool_params); in ConfigureInput()
610 CHECK_NOTNULL(dnn_data_input); in ConfigureInput()
645 CHECK_NOTNULL(output_tensor); in AllocateOutputTensor()
660 CHECK_NOTNULL(*output_tensor); in AllocateOutputTensor()
697 CHECK_NOTNULL(output_tensor); in AllocateOutputTensor()
711 CHECK_NOTNULL(*output_tensor); in AllocateOutputTensor()
|
| D | functional_ops.cc | 174 lib_(CHECK_NOTNULL(ctx_->function_library())) { in State()
263 lib_(CHECK_NOTNULL(ctx_->function_library())) { in State()
364 lib_(CHECK_NOTNULL(ctx_->function_library())) { in State()
523 lib_(CHECK_NOTNULL(ctx_->function_library())), in State()
|
| D | mkl_tfconv_op.h | 102 CHECK_NOTNULL(output_tensor); in ConvertMklToTf()
|
| D | lmdb_reader_op.cc | 99 CHECK_NOTNULL(mdb_cursor_); in Seek()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_tfconversion_pass.cc | 143 CHECK_NOTNULL(e); in InsertConversionNodeOnEdge()
148 CHECK_NOTNULL(src); in InsertConversionNodeOnEdge()
149 CHECK_NOTNULL(dst); in InsertConversionNodeOnEdge()
178 CHECK_NOTNULL(conversion_node); in InsertConversionNodeOnEdge()
197 CHECK_NOTNULL((*g)->AddEdge(conversion_node, 0, dst, e->dst_input())); in InsertConversionNodeOnEdge()
210 CHECK_NOTNULL(n); in InsertInputConversionNode()
246 CHECK_NOTNULL(conversion_node); in InsertInputConversionNode()
253 CHECK_NOTNULL((*g)->AddControlEdge(e->src(), conversion_node)); in InsertInputConversionNode()
281 CHECK_NOTNULL((*g)->AddEdge(conversion_node, 0, n, edges[0]->dst_input())); in InsertInputConversionNode()
282 CHECK_NOTNULL((*g)->AddEdge(conversion_node, 1, n, edges[1]->dst_input())); in InsertInputConversionNode()
[all …]
|
| D | mkl_layout_pass.cc | 861 CHECK_NOTNULL(m); in GetConv2DOrBiasAdd()
1018 CHECK_NOTNULL(m); in GetConv2DBackpropFilterOrBiasAddGrad()
1195 CHECK_NOTNULL(n); in NonDepthBatchWisePoolRewrite()
1221 CHECK_NOTNULL(n); in LrnRewrite()
1239 CHECK_NOTNULL(n); in LrnGradRewrite()
1280 CHECK_NOTNULL(n); in MaxpoolGradRewrite()
1298 CHECK_NOTNULL(n); in AddNRewrite()
1584 CHECK_NOTNULL(output_nodes); in GetNodesProducingTFTensorList()
1620 CHECK_NOTNULL(*out); // Make sure we got a valid object before using it in GetDummyMklTensorNode()
1637 CHECK_NOTNULL((*g)->AddControlEdge(orig_input0, *out, true)); in GetDummyMklTensorNode()
[all …]
|
| D | subgraph_test.cc | 210 EXPECT_FALSE(IsRefType(CHECK_NOTNULL(n)->output_type(0))); in TEST_F()
222 EXPECT_FALSE(IsRefType(CHECK_NOTNULL(n)->output_type(0))); in TEST_F()
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | logging_test.cc | 63 CHECK_NOTNULL(p_const); in TEST()
64 CHECK_NOTNULL(p_non_const); in TEST()
76 ASSERT_DEATH(printf("%s", CHECK_NOTNULL(p_null)), "Must be non NULL"); in TEST()
77 ASSERT_DEATH(printf("%s", CHECK_NOTNULL(p_null_const)), "Must be non NULL"); in TEST()
|
| D | file_system.cc | 77 CHECK_NOTNULL(undeleted_files); in DeleteRecursively()
78 CHECK_NOTNULL(undeleted_dirs); in DeleteRecursively()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 615 CHECK_NOTNULL(pd); in SetMklLayout()
620 CHECK_NOTNULL(md); in SetMklLayout()
906 CHECK_NOTNULL(input_tensors); in GetMklInputList()
946 CHECK_NOTNULL(context); in GetTfShape()
1685 CHECK_NOTNULL(tensor); in GetTensorBuffer()
1710 CHECK_NOTNULL(tensor); in SetUsrMem()
1742 CHECK_NOTNULL(tensor); in SetUsrMem()
1757 CHECK_NOTNULL(tensor); in SetUsrMem()
1767 CHECK_NOTNULL(cpu_engine_);
1780 CHECK_NOTNULL(tensor); in SetUsrMem()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | op_types.cc | 98 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsCastLike()
186 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsElementWiseMonotonic()
193 CHECK_NOTNULL((new gtl::FlatSet<string>{"Acos", "Erfc", "Neg", "Rsqrt"})); in IsElementWiseMonotonic()
517 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsTensorArray()
698 CHECK_NOTNULL((new gtl::FlatSet<string>{"Conj", "Reciprocal", "Invert", in OPDEF_PROPERTY_HELPER()
708 CHECK_NOTNULL((new const gtl::FlatSet<string>{ in IsValueAndOrderAndShapePreserving()
728 CHECK_NOTNULL((new const gtl::FlatSet<string>{ in IsValueAndOrderPreserving()
739 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsValuePreserving()
758 CHECK_NOTNULL((new gtl::FlatSet<string>{ in IsUnaryElementWise()
821 static const gtl::FlatSet<string>* const kNonForwardingOps = CHECK_NOTNULL( in NeverForwardsInputs()
|
| /external/tensorflow/tensorflow/core/lib/png/ |
| D | png_io.cc | 153 CHECK_NOTNULL(width); in DecodeHeader()
155 CHECK_NOTNULL(height); in DecodeHeader()
309 CHECK_NOTNULL(data); in CommonFinishDecode()
347 CHECK_NOTNULL(image); in WriteImageToBuffer()
348 CHECK_NOTNULL(png_string); in WriteImageToBuffer()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_util.cc | 37 CHECK_NOTNULL(reinterpret_cast<PyClassObject*>(py)->cl_name)); in ClassName()
39 return PyString_AS_STRING(CHECK_NOTNULL( in ClassName()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry_test.cc | 198 VariantValue* vv_out = CHECK_NOTNULL(v_out.get<VariantValue>()); in TEST()
224 VariantValue* vv_out = CHECK_NOTNULL(v_out.get<VariantValue>()); in TEST()
270 VariantValue* vv_out = CHECK_NOTNULL(v_out.get<VariantValue>()); in TEST()
297 VariantValue* vv_out = CHECK_NOTNULL(v_out.get<VariantValue>()); in TEST()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | ops.cc | 24 CHECK_NOTNULL(node_); in input()
41 CHECK_NOTNULL(node_); in output()
|
| D | gradients_test.cc | 95 CHECK_NOTNULL(z.node()); in TEST_F()
121 CHECK_NOTNULL(z.node()); in TEST_F()
151 CHECK_NOTNULL(z.node()); in TEST_F()
180 CHECK_NOTNULL(v.node()); in TEST_F()
186 CHECK_NOTNULL(z.node()); in TEST_F()
323 CHECK_NOTNULL(z.node()); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 168 CHECK_NOTNULL(node); in LoadGraphFromProto()
334 CHECK_NOTNULL(node); in TransformGraphToAddAggregatedInputNode()
353 CHECK_NOTNULL(input_node); in TransformGraphToAddAggregatedInputNode()
376 CHECK_NOTNULL(original_input_node); in TransformGraphToAddAggregatedInputNode()
382 CHECK_NOTNULL(created_node); in TransformGraphToAddAggregatedInputNode()
406 CHECK_NOTNULL(context); in TransformGraphToAddAggregatedInputNode()
685 CHECK_NOTNULL(input0_node); in RegisterNodeWithRank()
734 CHECK_NOTNULL(input_node); in RegisterPadNode()
739 CHECK_NOTNULL(tensor_proto); in RegisterPadNode()
875 CHECK_NOTNULL(input_node); in AddNodeInputByInputIndex()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_api_info_test.cc | 122 CHECK_NOTNULL(info); in GetInterfaceName()
129 CHECK_NOTNULL(info); in GetPreferredDevice()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 95 beam_scorer_(CHECK_NOTNULL(scorer)) { in CTCDecoder()
403 CHECK_NOTNULL(paths)->clear(); in TopPaths()
404 CHECK_NOTNULL(log_probs)->clear(); in TopPaths()
|
| /external/tensorflow/tensorflow/contrib/tensorrt/custom_plugin_examples/ |
| D | inc_op_kernel.cu.cc | 66 const cudaStream_t* stream = CHECK_NOTNULL( in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | execution_tracker.cc | 31 : backend_(CHECK_NOTNULL(backend)), in AsyncExecution()
|