Searched refs:check_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor_test.py | 66 def _assertStructuredEqual(self, a, b, msg, check_shape): argument
67 if check_shape:
75 self._assertStructuredEqual(a_value, b_value, msg, check_shape)
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_deprecated_py2.py | 138 name, check_shape, init, entry, exit_, shape_invariant): argument
168 if check_shape:
|
| D | control_flow.py | 213 name, check_shape, init, entry, exit_, shape_invariant): argument
246 if check_shape:
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 745 auto check_shape = in BuildTensor() local
764 if (mlir::failed(check_shape(shape_ref))) return llvm::None; in BuildTensor()
774 if (mlir::failed(check_shape(shape_ref))) return llvm::None; in BuildTensor()
779 if (mlir::failed(check_shape(shape_ref))) return llvm::None; in BuildTensor()
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1710 auto check_shape = [&](int input_index) -> Status { in GetArgAndRetvalShapes() local
1734 TF_RETURN_IF_ERROR(check_shape(j)); in GetArgAndRetvalShapes()
1739 TF_RETURN_IF_ERROR(check_shape(params_info.NumPerReplicaArgs() + i)); in GetArgAndRetvalShapes()
|