Searched refs:merged_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | beta_op.cc | 59 TensorShape merged_shape(a_shape); in Compile() local
60 if (b_shape.dims() > 0) merged_shape = b_shape; in Compile()
61 if (x_shape.dims() > 0) merged_shape = x_shape; in Compile()
67 auto a, BroadcastTo(ctx->Input(0), merged_shape.dim_sizes())); in Compile()
69 auto b, BroadcastTo(ctx->Input(1), merged_shape.dim_sizes())); in Compile()
71 auto x, BroadcastTo(ctx->Input(2), merged_shape.dim_sizes())); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | betainc_op.cc | 69 TensorShape merged_shape(a_shape); in Compute() local
70 if (b_shape.dims() > 0) merged_shape = b_shape; in Compute()
71 if (x_shape.dims() > 0) merged_shape = x_shape; in Compute()
74 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, merged_shape, &output)); in Compute()
83 auto merged_shape_vec = BCast::FromShape(merged_shape); in Compute()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 675 merged_shape = shape_list[0]
677 merged_shape.merge_with(s)
678 return merged_shape.as_list()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 2111 const auto& merged_shape = shape_manager->GetMergedShape(ic, shape); in ValidateSymbolicShapeManager() local
2112 if (!refiner->CompatibleShapes(shape, merged_shape)) { in ValidateSymbolicShapeManager()
2117 << " vs. merged: " << ic->DebugString(merged_shape); in ValidateSymbolicShapeManager()
2122 const auto& merged_shape = shape_manager->GetMergedShape(ic, shape); in ValidateSymbolicShapeManager() local
2123 if (!refiner->CompatibleShapes(shape, merged_shape)) { in ValidateSymbolicShapeManager()
2128 << " vs. merged: " << ic->DebugString(merged_shape); in ValidateSymbolicShapeManager()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1598 ShapeHandle merged_shape; in __anondb9326b22302() local
1600 c->Merge(desired_shape_handle, input_shape_handle, &merged_shape)); in __anondb9326b22302()
1601 c->set_output(0, merged_shape); in __anondb9326b22302()
|