Searched refs:init_shape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 80 TF_ASSIGN_OR_RETURN(Shape init_shape, b->GetShape(init_value)); in ComputeSums()
81 PrimitiveType accumulation_type = init_shape.element_type(); in ComputeSums()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 769 init_shape = None
771 init_shape = var_shape
774 init_shape = var_shape
779 init_shape = None
783 init_shape = None
788 shape=init_shape,
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | control_flow_deprecated_py2.py | 178 init_shape = init.shape
179 if not _is_subshape(init_shape, shape_invariant):
182 ' the shape invariant {}.'.format(name, init_shape,
|
| D | control_flow.py | 256 init_shape = init.shape
257 if not _is_subshape(init_shape, shape_invariant):
260 ' the shape invariant {}.'.format(name, init_shape,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 711 auto init_shape = while_inst->operand(0)->shape(); in CheckWhileLayout() local
714 init_shape, /*minor_to_major_only=*/true)); in CheckWhileLayout()
716 init_shape, /*minor_to_major_only=*/true)); in CheckWhileLayout()
718 init_shape, /*minor_to_major_only=*/true)); in CheckWhileLayout()
719 TF_RET_CHECK(LayoutsInShapesEqual(init_shape, while_inst->shape())); in CheckWhileLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2250 TF_ASSIGN_OR_RETURN(const Shape* init_shape, GetShapePtr(init)); in While()
2253 body_program_shape, *init_shape)); in While()
2564 TF_ASSIGN_OR_RETURN(const Shape* init_shape, GetShapePtr(init_value)); in ReduceWindowWithGeneralPadding()
2565 init_shapes.push_back(init_shape); in ReduceWindowWithGeneralPadding()
2608 TF_ASSIGN_OR_RETURN(const Shape* init_shape, GetShapePtr(init_value)); in ReduceWindowInternal()
2609 init_shapes.push_back(init_shape); in ReduceWindowInternal()
3081 TF_ASSIGN_OR_RETURN(const Shape* init_shape, GetShapePtr(init_value)); in SelectAndScatterInternal()
3093 *source_shape, *init_shape, scatter_shape)); in SelectAndScatterInternal()
|