Home
last modified time | relevance | path

Searched refs:ishape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dtranspose_benchmark.py131 for ishape, perm in zip(small_shapes, small_perms):
132 self._run_graph("gpu", ishape, perm, num_iters, datatype)
136 for ishape, perm in zip(large_shapes, large_perms):
137 self._run_graph("gpu", ishape, perm, num_iters, datatype)
148 for ishape, perm in zip(small_dim_large_shapes, small_dim_perms):
149 self._run_graph("gpu", ishape, perm, num_iters, datatype)
153 for ishape, perm in zip(small_dim_small_shapes, small_dim_perms):
154 self._run_graph("gpu", ishape, perm, num_iters, datatype)
Dconv2d_benchmark.py196 ishape = [batch_size, h, w, in_channel]
199 ishape = [batch_size, in_channel, h, w]
206 self._run_graph("gpu", dtype, data_format, ishape, fshape, stride,
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.cc229 TF_ASSIGN_OR_RETURN(const xla::Shape* ishape, in InputShapeMatches()
231 if (pshape.rank() != ishape->rank() || in InputShapeMatches()
232 pshape.element_type() != ishape->element_type()) { in InputShapeMatches()
235 if (pshape.is_static() && pshape.layout() != ishape->layout()) { in InputShapeMatches()
240 if (pshape.dimensions(dim) < ishape->dimensions(dim)) { in InputShapeMatches()
243 } else if (pshape.dimensions(dim) != ishape->dimensions(dim)) { in InputShapeMatches()
/external/tensorflow/tensorflow/python/keras/utils/
Dvis_utils.py220 [format_shape(ishape) for ishape in layer.input_shapes])
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py3464 ishape = [1, 400, 400, 1] variable
3469 GetInceptionFwdTest(ishape, fshape, 1, "SAME", gpu_only=True)))
3472 GetInceptionFwdDilatedConvTest(ishape, fshape, 1, "SAME")))
3475 GetInceptionBackInputTest(ishape, fshape, oshape, 1, "SAME",
3479 GetInceptionBackFilterTest(ishape, fshape, oshape, [1, 1], "SAME",
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc478 static int64 GetInferredShapeSize(const InferredShape& ishape, in GetInferredShapeSize() argument
480 return ishape.shape.IsFullyDefined() in GetInferredShapeSize()
481 ? ishape.shape.num_elements() * DataTypeSize(dtype) in GetInferredShapeSize()