| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | rsqrt_builder.cc | 55 int flat_shape[] = {1, 1, 1, num_elements_}; in AddNumerator() local
59 flat_shape, reinterpret_cast<char*>(numerator.data()), in AddNumerator()
91 int flat_shape[] = {1, 1, 1, num_elements_}; in PopulateSubGraph() local
110 reshape_shape, reinterpret_cast<char*>(flat_shape), in PopulateSubGraph()
111 sizeof(flat_shape[0]) * 4); in PopulateSubGraph()
116 auto reshape_out = reshape_op->AddOutput(sizeof(uint8_t), 4, flat_shape); in PopulateSubGraph()
147 auto div_output = div_op->AddOutput(sizeof(uint8_t), 4, flat_shape); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | scatter_nd_fuzz.cc | 66 auto flat_shape = shape_tensor.flat<int32>(); in FuzzImpl() local
68 flat_shape(i) = data[data_ix++] % kMaxDim; in FuzzImpl()
111 updates_dims.push_back(flat_shape(i)); in FuzzImpl()
112 num_indices *= flat_shape(i); in FuzzImpl()
|
| /external/tensorflow/tensorflow/python/data/util/ |
| D | structure.py | 164 for flat_type, flat_shape, flat_class in zip(flat_types, flat_shapes,
169 flat_ret.append(sparse_tensor.SparseTensorSpec(flat_shape, flat_type))
171 flat_ret.append(tensor_spec.TensorSpec(flat_shape, flat_type))
176 flat_shape[2:], flat_type,
177 dynamic_size=tensor_shape.dimension_value(flat_shape[0]),
178 infer_shape=tensor_shape.dimension_value(flat_shape[1])))
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | restore_ops.cc | 52 int64 flat_shape[] = {1}; in CreateStringVectorTensorHandle() local
53 AbstractTensorPtr tensor(ctx->CreateTensor(DT_STRING, flat_shape)); in CreateStringVectorTensorHandle()
|
| /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
73 auto flat = ops::Reshape(s.WithOpName("flat"), relu, flat_shape); in CreateMiniGraph()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_factory_ops.py | 344 flat_shape = tensor_shape.TensorShape([None]).concatenate(value_shape)
345 result = array_ops.placeholder(dtype, flat_shape, "flat_values")
|
| D | ragged_tensor.py | 1487 flat_shape = tensor_shape.as_shape([None] + shape[self.ragged_rank + 1:])
1488 self.flat_values.set_shape(flat_shape)
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_circulant.py | 250 flat_shape = vec_leading_shape + [np.prod(vec_block_shape)]
255 flat_shape = array_ops.concat(
257 vec_flat = array_ops.reshape(vec, flat_shape)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | keras_tensor.py | 454 flat_shape = ragged_spec.shape[ragged_spec.ragged_rank:]
455 result = array_ops.placeholder(ragged_spec.dtype, flat_shape)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | init_ops_v2.py | 683 flat_shape = (max(num_cols, num_rows), min(num_cols, num_rows))
686 a = self._random_generator.random_normal(flat_shape, dtype=dtype)
|
| D | init_ops.py | 582 flat_shape = (num_cols, num_rows)
584 flat_shape = (num_rows, num_cols)
587 a = random_ops.random_normal(flat_shape, dtype=dtype, seed=self.seed)
|
| D | array_ops.py | 5430 flat_shape = concat(([batch_size], indices_shape[batch_dims:]), axis=0)
5431 flat_indices = reshape(indices, shape=flat_shape)
|
| /external/tensorflow/tensorflow/python/keras/initializers/ |
| D | initializers_v2.py | 605 flat_shape = (max(num_cols, num_rows), min(num_cols, num_rows))
608 a = self._random_generator.random_normal(flat_shape, dtype=dtype)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier.cc | 895 const Shape& flat_shape = instr->shape(); in TryFlattenNestedTuples() local
896 for (int64 i = 0; i < flat_shape.tuple_shapes_size(); ++i) { in TryFlattenNestedTuples()
898 flat_shape.tuple_shapes(i), instr, i))); in TryFlattenNestedTuples()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/ |
| D | buffer_reuse.mlir | 354 %flat_shape : memref<1xindex>,
357 %flat_arg = memref_reshape %arg(%flat_shape)
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
| D | hlo-transform-unranked.mlir | 10 %flat_shape = tensor.from_elements %num_elements : tensor<1xindex>
11 %flat_a = "mhlo.dynamic_reshape"(%a, %flat_shape)
|
| /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/python/ops/numpy_ops/ |
| D | np_array_ops.py | 1768 flat_shape = array_ops.concat(
1771 tensor = array_ops.reshape(tensor, flat_shape)
|