Searched refs:flat_shape (Results 1 – 8 of 8) sorted by relevance
| /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/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/data/util/ |
| D | structure.py | 275 for flat_type, flat_shape, flat_class in zip(flat_types, flat_shapes,
280 flat_ret.append(SparseTensorStructure(flat_type, flat_shape))
282 flat_ret.append(TensorStructure(flat_type, flat_shape))
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_circulant.py | 251 flat_shape = vec_leading_shape + [np.prod(vec_block_shape)]
256 flat_shape = array_ops.concat(
258 vec_flat = array_ops.reshape(vec, flat_shape)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | init_ops_v2.py | 498 flat_shape = (max(num_cols, num_rows), min(num_cols, num_rows))
501 a = self._random_generator.random_normal(flat_shape, dtype=dtype)
|
| D | init_ops.py | 612 flat_shape = (num_cols, num_rows)
614 flat_shape = (num_rows, num_cols)
617 a = random_ops.random_normal(flat_shape, dtype=dtype, seed=self.seed)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier.cc | 683 const Shape& flat_shape = instr->shape(); in TryFlattenNestedTuples() local
684 for (int64 i = 0; i < flat_shape.tuple_shapes_size(); ++i) { in TryFlattenNestedTuples()
686 flat_shape.tuple_shapes(i), instr, i))); in TryFlattenNestedTuples()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | decode_proto_op.cc | 789 TensorShape flat_shape = {static_cast<int64>(message_count), in Compute() local
|