Home
last modified time | relevance | path

Searched refs:static_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dbatch_matmul_op_test.py58 def _compare(self, x_in, y_in, adjoint_a, adjoint_b, static_shape): argument
66 if static_shape:
86 static_shape=use_static_shape)
106 static_shape=use_static_shape)
124 static_shape=use_static_shape)
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc455 PartialTensorShape static_shape(arg.shape()); in ComputeArgumentShapes() local
458 if (static_shape.IsFullyDefined()) { in ComputeArgumentShapes()
459 TF_RET_CHECK(static_shape.AsTensorShape(&shape)); in ComputeArgumentShapes()
464 if (!static_shape.IsCompatibleWith(shape)) { in ComputeArgumentShapes()
468 static_shape.DebugString(), in ComputeArgumentShapes()
/external/tensorflow/tensorflow/python/ops/distributions/
Ddistribution.py549 static_shape = tensor_util.constant_value(shape)
550 if static_shape is None:
553 static_params[name] = tensor_shape.TensorShape(static_shape)
1312 def _is_scalar_helper(self, static_shape, dynamic_shape_fn): argument
1314 if static_shape.ndims is not None:
1315 return static_shape.ndims == 0
/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dutil_test.py709 static_shape = None if use_deferred_shape else x_.shape
710 x_pl = array_ops.placeholder_with_default(x_, shape=static_shape)
791 static_shape = None if use_deferred_shape else x_.shape
792 x_pl = array_ops.placeholder_with_default(x_, shape=static_shape)
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops.py590 static_shape = array_ops.broadcast_static_shape(a.get_shape(),
592 if static_shape.is_fully_defined():
593 output_shape = static_shape.as_list()
Dimage_ops_impl.py117 static_shape = image.get_shape().with_rank(rank).as_list()
120 s if s is not None else d for s, d in zip(static_shape, dynamic_shape)
925 static_shape = tensor.get_shape().dims[idx].value
926 if static_shape is not None:
927 return static_shape, False
Dcheck_ops.py1590 static_shape = x.get_shape().as_list()
1593 for i, size in enumerate(static_shape)
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1348 const Shape static_shape = xla::ShapeUtil::MakeStaticShape(current_shape); in GetOrCreateArrayView() local
1351 static_shape, builder_)); in GetOrCreateArrayView()
1370 static_shape); in GetOrCreateArrayView()
/external/python/cpython3/Modules/
D_testbuffer.c2696 static Py_ssize_t static_shape[1] = {12}; variable
2706 static_shape, /* shape */