Searched refs:base_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | helper.py | 341 base_shape = array_ops.shape(base_next_inputs)
344 shape=base_shape)
347 shape=base_shape))
475 base_shape = array_ops.shape(base_next_inputs)
478 shape=base_shape)
481 shape=base_shape))
|
| /external/tensorflow/tensorflow/contrib/gan/python/features/python/ |
| D | virtual_batchnorm_impl.py | 104 base_shape = tensor_shape.TensorShape(_get_shape(tensor_list[0]))
106 base_shape.assert_is_compatible_with(_get_shape(tensor))
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 247 StatusOr<Shape> InferWindowOutputShape(const Shape& base_shape, in InferWindowOutputShape() argument
251 if (window.dimensions_size() != ShapeUtil::Rank(base_shape)) { in InferWindowOutputShape()
254 window.dimensions_size(), ShapeUtil::Rank(base_shape)); in InferWindowOutputShape()
288 ShapeUtil::GetDimension(base_shape, i), dim.base_dilation()); in InferWindowOutputShape()
1728 Shape base_shape = in InferConvolveShape() local
1732 InferWindowOutputShape(base_shape, window, lhs.element_type(), in InferConvolveShape()
|
| D | hlo_evaluator.cc | 174 const Shape& window_shape, const Window& window, const Shape& base_shape, in IterateThroughWindow() argument
177 const int64 rank = ShapeUtil::Rank(base_shape); in IterateThroughWindow()
186 if (base_index[i] < 0 || base_index[i] >= base_shape.dimensions(i)) { in IterateThroughWindow()
|