Searched refs:init_shape (Results 1 – 5 of 5) 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 | 751 init_shape = None
753 init_shape = var_shape
756 init_shape = var_shape
761 init_shape = None
765 init_shape = None
770 shape=init_shape,
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 694 auto init_shape = while_inst->operand(0)->shape(); in CheckWhileLayout() local
697 init_shape)); in CheckWhileLayout()
699 init_shape)); in CheckWhileLayout()
701 body_computation_layout.result_layout().MatchesLayoutInShape(init_shape)); in CheckWhileLayout()
703 LayoutUtil::LayoutsInShapesEqual(init_shape, while_inst->shape())); in CheckWhileLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1828 TF_ASSIGN_OR_RETURN(const Shape& init_shape, GetShape(init)); in While()
1831 body_program_shape, init_shape)); in While()
2027 TF_ASSIGN_OR_RETURN(const Shape& init_shape, GetShape(init_value)); in ReduceWindowWithGeneralPadding()
2035 operand_shape, init_shape, in ReduceWindowWithGeneralPadding()
2272 TF_ASSIGN_OR_RETURN(const Shape& init_shape, GetShape(init_value)); in SelectAndScatterWithGeneralPadding()
2283 source_shape, init_shape, scatter_shape)); in SelectAndScatterWithGeneralPadding()
|
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| D | metric_ops.py | 3623 init_shape = [init_size] + fixed_shape
3625 init_shape, values.dtype, validate_shape=False, name='array')
|