Searched defs:flat_shape (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | rsqrt_builder.cc | 55 int flat_shape[] = {1, 1, 1, num_elements_}; in AddNumerator() local
91 int flat_shape[] = {1, 1, 1, num_elements_}; in PopulateSubGraph() local
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | restore_ops.cc | 52 int64 flat_shape[] = {1}; in CreateStringVectorTensorHandle() local
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | scatter_nd_fuzz.cc | 66 auto flat_shape = shape_tensor.flat<int32>(); in FuzzImpl() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | analytical_cost_estimator_test.cc | 71 auto flat_shape = ops::Const(s.WithOpName("flat_shape"), in CreateMiniGraph() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | rank_specialization.cc | 461 Value flat_shape = MaterializeFlatShape(b, loc, non_scalar_shapes); in MaterializeScalarRankSpecializationCase() local
525 Value flat_shape = MaterializeFlatShape(b, loc, non_scalar_shapes); in MaterializeEqualShapesRankSpecializationCase() local
695 Value flat_shape = MaterializeFlatShape(rewriter, loc, non_scalar_shapes); in MaterializeRankSpecializationForSingleNonScalarShapeEquivalenceClass() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | decode_proto_op.cc | 792 TensorShape flat_shape = {static_cast<int64>(message_count), in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier.cc | 933 const Shape& flat_shape = instr->shape(); in TryFlattenNestedTuples() local
|