Searched refs:is_tensor (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 389 if (!operand_l.is_tensor() && !operand_r.is_tensor()) { in GetTrtBroadcastShape()
403 if (use_implicit_batch && input.is_tensor()) { in GetTrtBroadcastShape()
433 (use_implicit_batch && operand_l.is_tensor()), in GetTrtBroadcastShape()
435 (use_implicit_batch && operand_r.is_tensor())); in GetTrtBroadcastShape()
609 bool is_tensor) { in DimsHaveSameSize() argument
610 if (is_tensor) { in DimsHaveSameSize()
619 const nvinfer1::Dims& rhs, bool is_tensor) { in AreDimsStaticWithSameSize() argument
621 return DimsHaveSameSize(lhs, rhs, is_tensor); in AreDimsStaticWithSameSize()
625 const nvinfer1::Dims& rhs, bool is_tensor) { in AreDimsStaticWithDifferentSize() argument
627 return !DimsHaveSameSize(lhs, rhs, is_tensor); in AreDimsStaticWithDifferentSize()
[all …]
|
| D | convert_nodes_test.cc | 460 EXPECT_EQ(false, ptr->is_tensor()); in TEST()
481 ASSERT_TRUE(ptr->is_tensor()); in TEST()
504 ASSERT_TRUE(ptr->is_tensor()); in TEST()
519 EXPECT_EQ(false, ptr->is_tensor()); in TEST()
610 ASSERT_TRUE(output.is_tensor()); in TEST_F()
1158 bool is_tensor, int batch_size = -1) { in TEST_F() argument
1159 if (is_tensor) { in TEST_F()
1685 ASSERT_TRUE(output.is_tensor()); in RunValidationAndConversion()
2403 ASSERT_TRUE(output.is_tensor()); in TEST_F()
2513 ASSERT_TRUE(output.is_tensor()); in TestMatMulHelper()
[all …]
|
| D | convert_nodes.h | 287 bool is_tensor() const { return initialized_ && is_tensor_; } in is_tensor() function
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | tensor_util.py | 875 if not is_tensor(tensor):
1065 is_tensor = is_tf_type variable
1116 not tensor.shape.is_fully_defined() and is_tensor(shape)):
|
| D | test_util.py | 371 is_tensor = isinstance(input_shape_or_tensor, ops.Tensor)
374 if is_tensor else input_shape_or_tensor)
382 if is_tensor:
405 is_tensor = isinstance(input_shape_or_tensor, ops.Tensor)
408 if is_tensor else input_shape_or_tensor)
414 if is_tensor:
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.pbtxt | 752 name: "is_tensor"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 1536 name: "is_tensor"
|
| /external/tensorflow/ |
| D | RELEASE.md | 376 * Code that uses `isinstance(x, tf.Tensor)` instead of `tf.is_tensor` when
377 checking Keras symbolic inputs/outputs should switch to using `tf.is_tensor`.
|