Home
last modified time | relevance | path

Searched refs:original_shape (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreplay_test.cc54 std::unique_ptr<ProgramShape> original_shape = in TEST_F() local
58 ASSERT_TRUE(protobuf_util::ProtobufEquals(original_shape->ToProto(), in TEST_F()
87 std::unique_ptr<ProgramShape> original_shape = in XLA_TEST_F() local
91 ASSERT_TRUE(protobuf_util::ProtobufEquals(original_shape->ToProto(), in XLA_TEST_F()
134 std::unique_ptr<ProgramShape> original_shape = in TEST_F() local
138 ASSERT_TRUE(protobuf_util::ProtobufEquals(original_shape->ToProto(), in TEST_F()
/external/tensorflow/tensorflow/contrib/periodic_resample/kernels/
Dperiodic_resample_op.h269 const tensorflow::TensorShape& original_shape, in do_periodic_resample_op() argument
286 process_desired_shape(context, original_shape, desired_shape.dim_sizes(), in do_periodic_resample_op()
302 output_shape = original_shape; in do_periodic_resample_op()
315 InputIndexer input_indexer(target_dimensions, original_shape, in do_periodic_resample_op()
368 const tensorflow::TensorShape& original_shape, in create_grad_tensor() argument
373 context, original_shape, desired_shape, grad_tensor); \ in create_grad_tensor()
408 context->GetAttr("original_shape", &original_shape)); in PeriodicResampleOpGrad()
415 create_grad_tensor(context, grad_tensor, grad_tensor_type, original_shape, in Compute()
420 tensorflow::TensorShape original_shape;
/external/tensorflow/tensorflow/contrib/periodic_resample/ops/
Darray_ops.cc147 tensorflow::TensorShape original_shape; in __anon96b295990202() local
148 TF_RETURN_IF_ERROR(c->GetAttr("original_shape", &original_shape)); in __anon96b295990202()
150 TF_RETURN_IF_ERROR(c->MakeShapeFromTensorShape(original_shape, &s)); in __anon96b295990202()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmirror_pad_op.cc30 const xla::Shape& original_shape, in DoMirrorPad() argument
41 for (int64 dimno = original_shape.rank() - 1; dimno >= 0; --dimno) { in DoMirrorPad()
45 int64 dim_size = original_shape.dimensions(dimno); in DoMirrorPad()
/external/tensorflow/tensorflow/python/data/experimental/ops/
Dscan_ops.py109 for original_shape, weakened_shape in zip(flat_state_shapes,
111 if original_shape.ndims is not None and (
113 original_shape.as_list() != weakened_shape.as_list()):
Dgrouping.py334 for original_shape, weakened_shape in zip(flat_state_shapes,
336 if original_shape.ndims is not None and (
338 original_shape.as_list() != weakened_shape.as_list()):
Dbatching.py539 for original_shape, new_shape in zip(flat_original_shapes,
541 if not original_shape.is_compatible_with(new_shape):
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dreshape_vectorizer.cc34 Output GetVectorizedShape(Scope* s, Output tensor, Output original_shape) { in GetVectorizedShape() argument
44 return ops::Concat(*s, {dim_0, original_shape}, ops::Const(*s, 0)); in GetVectorizedShape()
/external/tensorflow/tensorflow/contrib/data/python/ops/
Dbatching.py204 original_shape.merge_with(new_shape)
205 for original_shape, new_shape in zip(flat_original_shapes,
/external/tensorflow/tensorflow/python/ops/
Dembedding_ops.py762 original_shape = sparse_ids.dense_shape
766 array_ops.size(original_shape)
771 array_ops.slice(original_shape, [0], [original_rank - 1])),
772 array_ops.gather(original_shape, original_rank - 1)])
817 math_ops.cast(original_shape, dtypes.int32), [0],
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Dbatch_reshape.py373 def calculate_reshape(original_shape, new_shape, validate=False, name=None): argument
378 with ops.name_scope(name, "calculate_reshape", [original_shape, new_shape]):
379 original_size = math_ops.reduce_prod(original_shape)
388 original_shape, 1, message="Original shape must be a vector."),
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dresnet_preprocessing.py96 original_shape = tf.image.extract_jpeg_shape(image_bytes)
97 bad = _at_least_x_are_equal(original_shape, tf.shape(image), 3)
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_shape_test.py303 original_shape = RaggedTensorDynamicShape.from_dim_sizes(original_dim_sizes)
305 self.assertEqual(original_shape.rank, bcast_shape.rank)
307 bcast1 = original_shape.broadcast_dimension(axis, row_length)
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dembedding_ops.py126 original_shape = sparse_ids.dense_shape
130 array_ops.size(original_shape)
135 array_ops.slice(original_shape, [0], [original_rank - 1])),
136 array_ops.gather(original_shape, original_rank - 1)])
181 math_ops.cast(original_shape, dtypes.int32), [0],
Dlayers.py268 original_shape = inputs.get_shape()
270 original_rank = original_shape.ndims
422 if original_shape.ndims == 2:
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_normalization.cc237 auto original_shape = hlo->shape(); in HandleMultipleOutputs() local
265 *tuple->mutable_shape() = original_shape; in HandleMultipleOutputs()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.h29 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape,
Dop_level_cost_estimator.cc183 TensorShapeProto MaybeGetMinimumShape(const TensorShapeProto& original_shape, in MaybeGetMinimumShape() argument
185 auto shape = original_shape; in MaybeGetMinimumShape()
203 shape.add_dim()->set_size(original_shape.dim(i).size()); in MaybeGetMinimumShape()
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization.py576 original_shape = [-1] + inputs.shape.as_list()[1:]
577 expanded_shape = [self.virtual_batch_size, -1] + original_shape[1:]
583 outputs = array_ops.reshape(outputs, original_shape)
/external/tensorflow/tensorflow/python/kernel_tests/
Darray_ops_test.py946 original_shape = constant_op.constant([6, 4, 4], dtype=dtypes.int64)
951 dx = array_ops.strided_slice_grad(original_shape, begin, end, strides,
961 original_shape = constant_op.constant([6, 4, 4], dtype=dtypes.int64)
969 dx = array_ops.strided_slice_grad(original_shape, begin, end, strides,
/external/tensorflow/tensorflow/python/keras/
Dbackend.py2409 original_shape = int_shape(x)
2426 if original_shape[rows] is None:
2429 new_height = original_shape[rows] * height_factor
2431 if original_shape[cols] is None:
2434 new_width = original_shape[cols] * width_factor
/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc333 TensorShape original_shape = old_lhs->shape(); in Compute() local
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py1323 for original_shape, weakened_shape in zip(flat_state_shapes,
1325 if original_shape.ndims is not None and (
1327 original_shape.as_list() != weakened_shape.as_list()):
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc935 TF_ASSIGN_OR_RETURN(const Shape& original_shape, GetShape(operand)); in Collapse()
937 VLOG(3) << "original shape: " << ShapeUtil::HumanString(original_shape); in Collapse()
941 for (int i = 0; i < original_shape.rank(); ++i) { in Collapse()
943 new_sizes.push_back(original_shape.dimensions(i)); in Collapse()
945 new_sizes.back() *= original_shape.dimensions(i); in Collapse()