- Sort Score
- Result 10 results
- Languages All
Results 11 - 19 of 19 for DT_INT64 (0.12 sec)
-
tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc
return DT_UINT16; case toco::IODataType::INT32: return DT_INT32; case toco::IODataType::UINT32: return DT_UINT32; case toco::IODataType::INT64: return DT_INT64; case toco::IODataType::UINT8: return DT_UINT8; case toco::IODataType::UINT64: return DT_UINT64; case toco::IODataType::STRING: return DT_STRING; case toco::IODataType::BOOL:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 17.3K bytes - Viewed (0) -
tensorflow/compiler/jit/node_matchers.cc
case DT_INT16: return CompareTensor<int16>(tensor, expected_tensor, listener); case DT_INT32: return CompareTensor<int32>(tensor, expected_tensor, listener); case DT_INT64: return CompareTensor<int64_t>(tensor, expected_tensor, listener); case DT_UINT8: return CompareTensor<uint8>(tensor, expected_tensor, listener); case DT_UINT16:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 03 16:15:20 UTC 2022 - 16.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-prefer-tf2xla.mlir
%output = "tf.StridedSlice"(%input,...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 06 15:32:52 UTC 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/internal/saved_model_api_test.cc
ASSERT_EQ(absl::OkStatus(), model_api->GetVariable("sub_module.uninitialized_variable", &uninitialized_variable)); ASSERT_EQ(tensorflow::DT_INT64, uninitialized_variable->dtype()); TF_DeleteSavedModel(saved_model); TF_DeleteStatus(status); TFE_DeleteContext(ctx); } TEST_P(CSavedModelAPITest, LoadSavedModelWithWhileLoop) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 08:08:45 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad_test.cc
} TEST_F(ArrayGradTest, GatherNdGrad_SliceIndexing_Int64) { TensorShape shape({2, 2}); auto x = Placeholder(scope_, DT_FLOAT, Placeholder::Shape(shape)); auto indices = Cast(scope_, Const(scope_, {{1}, {0}}), DT_INT64); auto y = GatherNd(scope_, x, indices); RunTest(x, shape, y, shape); } TEST_F(ArrayGradTest, CheckNumericsGrad) { TensorShape shape({5, 2});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 19.3K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_util.cc
case DT_INT16: case DT_QINT16: return PrintArray(num_elts, t.flat<int16>().data()); case DT_INT8: case DT_QINT8: return PrintArray(num_elts, t.flat<int8>().data()); case DT_INT64: return PrintArray(num_elts, t.flat<int64_t>().data()); case DT_BOOL: return PrintArray(num_elts, t.flat<bool>().data()); case DT_STRING: { string ret;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 25K bytes - Viewed (0) -
tensorflow/cc/gradients/nn_grad.cc
bool overlapping; TF_RETURN_IF_ERROR( GetNodeAttr(op.output(0).node()->attrs(), "overlapping", &overlapping)); auto dx = internal::FractionalAvgPoolGrad( scope, Shape(scope, op.input(0), Shape::OutType(DT_INT64)), grad_inputs[0], op.output(1), op.output(2), internal::FractionalAvgPoolGrad::Overlapping(overlapping)); grad_outputs->push_back(dx); return scope.status(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 27 23:34:33 UTC 2022 - 24.5K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
// so it's fine to convert it back to int32 regardless of truncation. auto params = op.input(0); auto colocate_scope = scope.ColocateWith(params); Shape::Attrs shape_attrs; shape_attrs.out_type_ = DT_INT64; auto params_shape64 = Shape(colocate_scope, params, shape_attrs); Output params_shape = Cast(colocate_scope, params_shape64, DT_INT32); auto indices = op.input(1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
ReplaceAllUsesWithCast(concat.getValue(), buffer); // Create the lengths as a list of the same value (element size). tensorflow::Tensor lengths_tensor(tensorflow::DT_INT64, {buffer_type.getDimSize(0)}); for (int64_t i = 0; i < buffer_type.getDimSize(0); ++i) { lengths_tensor.vec<int64_t>()(i) = buffer_type.getDimSize(1); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0)